Correct cannot collect test class warnings

Signed-off-by: Gábor Lipták <[email protected]>
capitalone · Jun 16, 2024 · 3fbd8fc · 3fbd8fc
1 parent 4e4450a
commit 3fbd8fc
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 2 deletions.
diff --git a/dataprofiler/tests/profilers/test_histogram_utils.py b/dataprofiler/tests/profilers/test_histogram_utils.py
@@ -33,6 +33,8 @@ def mock_sqrt_return_nan(profile):
 
 
 class TestColumn(NumericStatsMixin):
+    __test__ = False
+
     def __init__(self):
         NumericStatsMixin.__init__(self)
         self.times = defaultdict(float)

diff --git a/dataprofiler/tests/profilers/test_numeric_stats_mixin_profile.py b/dataprofiler/tests/profilers/test_numeric_stats_mixin_profile.py
@@ -19,6 +19,8 @@
 
 
 class TestColumn(NumericStatsMixin):
+    __test__ = False
+
     def __init__(self):
         NumericStatsMixin.__init__(self)
         self.match_count = 0
@@ -32,6 +34,7 @@ def _filter_properties_w_options(self, calculations, options):
 
 
 class TestColumnWProps(TestColumn):
+    __test__ = False
     # overrides the property func
     median = None
     mode = None
@@ -52,7 +55,6 @@ class TestNumericStatsMixin(unittest.TestCase):
         create=True,
     )
     def test_base(self):
-
         # validate requires NumericalOptions
         with self.assertRaisesRegex(
             ValueError,
@@ -345,7 +347,6 @@ def test_timeit(self):
 
         time_array = [float(i) for i in range(24, 0, -1)]
         with mock.patch("time.time", side_effect=lambda: time_array.pop()):
-
             # Validate that the times dictionary is empty
             self.assertEqual(defaultdict(float), num_profiler.times)