Update test_analysis.py

TAKE 2
TheBattCave · Jun 6, 2024 · c1e209b · c1e209b
1 parent 25ec302
commit c1e209b
Showing 1 changed file with 5 additions and 4 deletions.
diff --git a/alfred/tests/test_analysis.py b/alfred/tests/test_analysis.py
@@ -1,4 +1,5 @@
 import alfred
+import unittest
 import altair as alt
 import csv
 import matplotlib.pyplot as plt
@@ -15,10 +16,10 @@
 data_path = os.path.join(alfred.__path__[0], 'data/sorted_test_data')
 
 class TestCountSwappedModules(unittest.TestCase):
-  def setUp(self):
-    self.valid_directory = os.path.join(data_path, 'all_buses/')
+    def setUp(self):
+        self.valid_directory = os.path.join(data_path, 'all_buses/')
 
-  def test_count_swapped_modules_output(self):
+    def test_count_swapped_modules_output(self):
         # Capture the output of the function
         captured_output = StringIO()
         sys.stdout = captured_output
@@ -38,7 +39,7 @@ def test_count_swapped_modules_output(self):
         self.assertIn("Count of '1':", captured_output_str)
         self.assertIn("0 = healthy and 1 = swapped", captured_output_str)
     def tearDown(self):
-      pass
+         pass
 
 if __name__ == '__main__':
     unittest.main()