####################################################
# CS 5000: F23: Assignment 2: Unit Tests
# bugs to vladimir kulyukin in canvas
#####################################################

import numpy as np
from Mod3DFA import Mod3DFA
from Mod7DFA import Mod7DFA
import unittest


class cs5000_f23_hw02_uts(unittest.TestCase):
    def test_assgn_02_ut_01(self):
        print("\n***** Testing Mod3DFA: Hw02: UT01 ************")
        lower, upper = 0, 10001
        dfa = Mod3DFA()
        for i in range(lower, upper):
            bstr = "{0:b}".format(i)
            if i % 3 == 0:
                assert dfa.processString(bstr) == True
                # print('assertion passed on {}'.format(i))
            else:
                assert dfa.processString(bstr) == False
                # print('assertion passed on {}'.format(i))
        print("Mod7DFA: HW02: UT01 passed...")

    def test_assgn_02_ut_02(self):
        print("\n***** Mod7DFA: HW03: UT02 ************")
        lower, upper = 0, 1000001
        dfa = Mod7DFA()
        for i in range(lower, upper):
            bstr = "{0:b}".format(i)
            if i % 7 == 0:
                assert dfa.processString(bstr) == True
                # print('assertion passed on {}'.format(i))
            else:
                assert dfa.processString(bstr) == False
                # print('assertion passed on {}'.format(i))
        print("Mod7DFA: HW02: UT02: passed...")


### ================ Unit Tests ====================

if __name__ == "__main__":
    unittest.main()