proposed fix to #39244 (#3)

* proposed fix to #39244 * Update test_module.py
4 years ago · 5a2c48afe4
parent 3f27096a3e
commit 5a2c48afe4
1 changed files with 1 additions and 1 deletions
--- a/test_module.py
+++ b/test_module.py
@ -4,7 +4,7 @@ import matplotlib as mpl

 class DataCleaningTestCase(unittest.TestCase):
    def test_data_cleaning(self):
-        actual = int(time_series_visualizer.df.count())
+        actual = int(time_series_visualizer.df.count(numeric_only=True))
        expected = 1238
        self.assertEqual(actual, expected, "Expected DataFrame count after cleaning to be 1238.")