diff --git a/tests/test_models.py b/tests/test_models.py index 4a9b6359b..1a37d421f 100644 --- a/tests/test_models.py +++ b/tests/test_models.py @@ -41,6 +41,12 @@ def test_max_mag(test_df, test_colname, expected): columns=list("abc")), "b", 0), + (pd.DataFrame(data=[[-7, -7, -3], + [-4, -3, -1], + [-1, -5, -3]], + columns=list("abc")), + "a", + -4), ]) def test_mean_mag(test_df, test_colname, expected): """Test mean function works for array of zeroes and positive integers."""