From 018a8811cd1737e5811ebaceb65de2ccdc7b564f Mon Sep 17 00:00:00 2001
From: Arunjith <118789403+arunjith-official@users.noreply.github.com>
Date: Thu, 12 Sep 2024 22:17:47 +0530
Subject: [PATCH] Update test_module.py

Instead of converting the whole DataFrame, adding the code to count the number of rows or columns
---
 test_module.py | 5 ++---
 1 file changed, 2 insertions(+), 3 deletions(-)

diff --git a/test_module.py b/test_module.py
index 51b094643..b1fe8012b 100644
--- a/test_module.py
+++ b/test_module.py
@@ -4,10 +4,9 @@
 
 class DataCleaningTestCase(unittest.TestCase):
     def test_data_cleaning(self):
-        actual = int(time_series_visualizer.df.count(numeric_only=True))
+        actual = len(time_series_visualizer.df)
         expected = 1238
-        self.assertEqual(actual, expected, "Expected DataFrame count after cleaning to be 1238.")
-
+        self.assertEqual(actual, expected,"Expected DataFrame count after cleaning to be 1238.")
 class LinePlotTestCase(unittest.TestCase):
     def setUp(self):
         self.fig = time_series_visualizer.draw_line_plot()