waka-readme/tests/test_main.py

'''
Tests for the main.py
'''
from main import make_graph
import unittest


class TestMain(unittest.TestCase):

    def test_make_graph(self):
        '''Tests the make_graph function'''
        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__':
    unittest.main()