transetting_test.py

import transettings
import ruleset
import propset
import unittest

class Test_TranSettings(unittest.TestCase):
    def test_random_region(self):
        region1 = transettings.random_region()
        region2 = transettings.random_region()
        print(f"{region1} vs {region2}")
        self.assertNotEqual(region1, region2)

TRANS_TICKET_NUM = 10

class Test_TransGenerator(unittest.TestCase):
    def test_gen(self):
        tg = transettings.TransGenerator()
        arr = tg.generate_settings(TRANS_TICKET_NUM)
        self.assertEqual(len(arr), TRANS_TICKET_NUM)

    def test_check(self):
        tg = transettings.TransGenerator()
        arr = tg.generate_settings(TRANS_TICKET_NUM)

        resarr = []
        checker = ruleset.RuleChecker()
        for ts in arr:
            tp = propset.TransProps(ts)
            res = checker.judge(tp)
            resarr.append(res)
        self.assertEqual(len(resarr), TRANS_TICKET_NUM)

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