diff --git a/source/tests/pt/test_stat.py b/source/tests/pt/test_stat.py index a3a29d74b9..b7e3f6e2d3 100644 --- a/source/tests/pt/test_stat.py +++ b/source/tests/pt/test_stat.py @@ -337,7 +337,6 @@ def tf_compute_input_stats(self): ) - class TestExcludeTypes(DatasetTest, unittest.TestCase): def setup_data(self): original_data = str(Path(__file__).parent / "water/data/data_0") @@ -375,8 +374,7 @@ def tf_compute_input_stats(self): box = self.dp_merged["box"] self.dp_d.compute_input_stats(coord, box, atype, natoms, self.dp_mesh, {}) - - + class TestOutputStat(unittest.TestCase): def setUp(self): self.data_file = [str(Path(__file__).parent / "water/data/data_0")] @@ -465,6 +463,5 @@ def test_assigned(self): ) - if __name__ == "__main__": unittest.main()