refactor: remove duplicate codes in the tests

This commit is contained in:
Yufan You 2020-08-16 16:08:06 +08:00
parent d827cb172d
commit 215367a2ad
No known key found for this signature in database
GPG Key ID: 863A0F9FA8127FA4

View File

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