python
/
nyaggle
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

add assert test

pull/81/head
wakamezake 2020-09-03 14:28:06 +09:00
parent 3072e3f25d
commit 133dab4813
1 changed files with 7 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
tests/feature/test_groupby.py
View File

@ -28,3 +28,10 @@ def test_return_type_by_aggregation(iris_dataframe):
agg_methods)
assert isinstance(new_df, pd.DataFrame)
assert isinstance(new_cols, list)
@pytest.mark.parametrize('agg_method', [[int], [lambda x: np.max(x)]])
def test_assert_by_aggregation(iris_dataframe, agg_method):
df, group_key, group_values = iris_dataframe
with pytest.raises(ValueError):
aggregation(df, group_key, group_values, agg_method)