Feature/black (#40)

* linted tmle_core * enhanced mediation tests
ronikobrosly · Apr 25, 2021 · bb9dab9 · bb9dab9
1 parent eb0e826
commit bb9dab9
Showing 1 changed file with 5 additions and 4 deletions.
diff --git a/tests/unit/test_mediation.py b/tests/unit/test_mediation.py
@@ -52,17 +52,17 @@ def test_mediation_fit(mediation_fixture):
         (0, 0.01, 0.99, 10, 10, 3, 5, 0.5, 100, None, True),
         (1e6, 0.01, 0.99, 10, 10, 3, 5, 0.5, 100, None, True),
         (10, "hehe", 0.99, 10, 10, 3, 5, 0.5, 100, None, True),
-        (10, -1, 0.99, 10, 10, 3, 5, 0.5, 100, None, True),
+        (10, -1.0, 0.99, 10, 10, 3, 5, 0.5, 100, None, True),
         (10, 1.5, 0.99, 10, 10, 3, 5, 0.5, 100, None, True),
         (10, 0.1, "hehe", 10, 10, 3, 5, 0.5, 100, None, True),
-        (10, 0.1, -1, 10, 10, 3, 5, 0.5, 100, None, True),
+        (10, 0.1, -1.0, 10, 10, 3, 5, 0.5, 100, None, True),
         (10, 0.1, 1.5, 10, 10, 3, 5, 0.5, 100, None, True),
         (10, 0.1, 0.9, 10.5, 10, 3, 5, 0.5, 100, None, True),
         (10, 0.1, 0.9, -2, 10, 3, 5, 0.5, 100, None, True),
-        (10, 0.1, 0.9, 1e6, 10, 3, 5, 0.5, 100, None, True),
+        (10, 0.1, 0.9, 10000000, 10, 3, 5, 0.5, 100, None, True),
         (10, 0.1, 0.9, 100, "10", 3, 5, 0.5, 100, None, True),
         (10, 0.1, 0.9, 100, -1, 3, 5, 0.5, 100, None, True),
-        (10, 0.1, 0.9, 100, 1e6, 3, 5, 0.5, 100, None, True),
+        (10, 0.1, 0.9, 100, 10000000, 3, 5, 0.5, 100, None, True),
         (10, 0.1, 0.9, 100, 200, "3", 5, 0.5, 100, None, True),
         (10, 0.1, 0.9, 100, 200, 1, 5, 0.5, 100, None, True),
         (10, 0.1, 0.9, 100, 200, 1e6, 5, 0.5, 100, None, True),
@@ -77,6 +77,7 @@ def test_mediation_fit(mediation_fixture):
         (10, 0.1, 0.9, 100, 200, 5, 10, 1, 100, "None", True),
         (10, 0.1, 0.9, 100, 200, 5, 10, 1, 100, -5, True),
         (10, 0.1, 0.9, 100, 200, 5, 10, 1, 100, 123, "True"),
+        (10, 30.0, 10.0, 100, 200, 5, 10, 1, 100, None, True),
     ],
 )
 def test_bad_mediation_instantiation(