From bb9dab9362bbf88aef42967d023ed9ceee3672fd Mon Sep 17 00:00:00 2001 From: Roni Kobrosly Date: Sun, 25 Apr 2021 19:58:09 -0400 Subject: [PATCH] Feature/black (#40) * linted tmle_core * enhanced mediation tests --- tests/unit/test_mediation.py | 9 +++++---- 1 file changed, 5 insertions(+), 4 deletions(-) diff --git a/tests/unit/test_mediation.py b/tests/unit/test_mediation.py index aa238c6..6fcac29 100644 --- 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(