diff --git a/jass/test/test_init_table.py b/jass/test/test_init_table.py index 3d7b0ef77d96b29fa17daef24032b2fd5c8de1df..bb400510c1f734cb3e78c1f9846f695a62fe7e1f 100644 --- a/jass/test/test_init_table.py +++ b/jass/test/test_init_table.py @@ -12,7 +12,7 @@ from . import JassTestCase class TestInitTable(object): - + test_folder = "data_real" __test__ = False def setUp(self): @@ -69,16 +69,13 @@ class TestInitTable(object): assert_frame_equal(self.expected_cov, self.result_cov, check_like=True) -params = [("nonans", "data_real", True), ("withnans", "data_real", False)] +class TestInitTableNoNans(TestInitTable, JassTestCase): + computed_cov = True + + +class TestInitTableWithNans(TestInitTable, JassTestCase): + computed_cov = False -for name, param, computed_cov in params: - cls_name = "TestInitTable_%s" % (name,) - print(cls_name) - globals()[cls_name] = type( - cls_name, - (TestInitTable, JassTestCase), - {"test_folder": param, "computed_cov":computed_cov, "__test__": True}, - ) if __name__ == "__main__": import unittest