refactor: remove duplicate codes in the tests

Author: ouuan

tests/test_main.py

@@ -9,30 +9,20 @@ class TestMain(unittest.TestCase):
 
     def test_make_graph(self):
         '''Tests the make_graph function'''
-        self.assertEqual(make_graph(0), "░░░░░░░░░░░░░░░░░░░░░░░░░",
-                         "0% should return ░░░░░░░░░░░░░░░░░░░░░░░░░")
-        self.assertEqual(make_graph(100), "█████████████████████████",
-                         "100% should return █████████████████████████")
-        self.assertEqual(make_graph(50), "████████████▒░░░░░░░░░░░░",
-                         "50% should return ████████████▒░░░░░░░░░░░░")
-        self.assertEqual(make_graph(50.001), "████████████▓░░░░░░░░░░░░",
-                         "50.001% should return ████████████▓░░░░░░░░░░░░")
-        self.assertEqual(make_graph(25), "██████▒░░░░░░░░░░░░░░░░░░",
-                         "25% should return ██████▒░░░░░░░░░░░░░░░░░░")
-        self.assertEqual(make_graph(75), "██████████████████▓░░░░░░",
-                         "75% should return ██████████████████▓░░░░░░")
-        self.assertEqual(make_graph(3.14), "▓░░░░░░░░░░░░░░░░░░░░░░░░",
-                         "3.14% should return ▓░░░░░░░░░░░░░░░░░░░░░░░░")
-        self.assertEqual(make_graph(9.901), "██▒░░░░░░░░░░░░░░░░░░░░░░",
-                         "9.901% should return ██▒░░░░░░░░░░░░░░░░░░░░░░")
-        self.assertEqual(make_graph(87.334), "██████████████████████░░░",
-                         "87.334% should return ██████████████████████░░░")
-        self.assertEqual(make_graph(87.333), "█████████████████████▓░░░",
-                         "87.333% should return █████████████████████▓░░░")
-        self.assertEqual(make_graph(4.666), "█░░░░░░░░░░░░░░░░░░░░░░░░",
-                         "4.666% should return █░░░░░░░░░░░░░░░░░░░░░░░░")
-        self.assertEqual(make_graph(4.667), "█▒░░░░░░░░░░░░░░░░░░░░░░░",
-                         "4.667% should return █▒░░░░░░░░░░░░░░░░░░░░░░░")
+        def test(percent: float, result: str):
+            self.assertEqual(make_graph(percent), result, f"{percent}% should return {result}")
+        test(0, "░░░░░░░░░░░░░░░░░░░░░░░░░")
+        test(100, "█████████████████████████")
+        test(50, "████████████▒░░░░░░░░░░░░")
+        test(50.001, "████████████▓░░░░░░░░░░░░")
+        test(25, "██████▒░░░░░░░░░░░░░░░░░░")
+        test(75, "██████████████████▓░░░░░░")
+        test(3.14, "▓░░░░░░░░░░░░░░░░░░░░░░░░")
+        test(9.901, "██▒░░░░░░░░░░░░░░░░░░░░░░")
+        test(87.334, "██████████████████████░░░")
+        test(87.333, "█████████████████████▓░░░")
+        test(4.666, "█░░░░░░░░░░░░░░░░░░░░░░░░")
+        test(4.667, "█▒░░░░░░░░░░░░░░░░░░░░░░░")
 
 
 if __name__ == '__main__':