Correct cannot collect test class warnings

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

Hide file tree

Showing 2 changed files with 3 additions and 0 deletions.
diff --git a/dataprofiler/tests/profilers/test_histogram_utils.py b/dataprofiler/tests/profilers/test_histogram_utils.py
@@ -33,6 +33,7 @@ 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,7 @@
 
 
 class TestColumn(NumericStatsMixin):
+    __test__ = False
     def __init__(self):
         NumericStatsMixin.__init__(self)
         self.match_count = 0
@@ -32,6 +33,7 @@ def _filter_properties_w_options(self, calculations, options):
 
 
 class TestColumnWProps(TestColumn):
+    __test__ = False
     # overrides the property func
     median = None
     mode = None