There was an error while loading. Please reload this page.
1 parent 7e65912 commit db18775Copy full SHA for db18775
1 file changed
codonPython/tests/suppression_test.py
@@ -0,0 +1,21 @@
1
+# from codonpython.suppression import suppression
2
+import pandas as pd
3
+import pytest
4
+
5
+@pytest.mark.parametrize("to_suppress, expected", [
6
+ (
7
+ pd.DataFrame({"count":[0,2,4,6,8,10,101,1001]}),
8
+ pd.DataFrame({"count":[0,"*","*","*",10,10,100,1000]})
9
+ )
10
+])
11
+def test_supression_BAU(to_generate, random_state, expected):
12
+ assert expected == suppression(to_generate, random_state=random_state)
13
14
+@pytest.mark.parametrize("to_suppress", [
15
+ pd.DataFrame({"count":[0,-2,4,6,8,10,101,1001]}),
16
+ pd.DataFrame({"count":[0,2,4,6,8,10,101,100000000]}),
17
+ pd.DataFrame({"count":[0,2.2,4,6,8,10,101,1001]})
18
19
+def test_supression_valueErrors(to_generate):
20
+ with pytest.raises(ValueError):
21
+ suppression(to_generate)
0 commit comments