class TestIntegerGenerator(unittest.TestCase): def setUp(self): self.integer_generator = IntegerGenerator() def test_generate_integer(self): generated = self.integer_generator.generate() self.assertEqual(type(generated), int) self.assertGreaterEqual(generated, 1) self.assertLessEqual(generated, 100) def test_generate_even_integer(self): self.integer_generator.even = IntegerGenerator.EVEN for i in range(20): generated = self.integer_generator.generate() self.assertEqual(type(generated), int) self.assertGreaterEqual(generated, 1) self.assertLessEqual(generated, 100) self.assertEqual(generated % 2, 0) def test_generate_odd_integer(self): self.integer_generator.even = IntegerGenerator.ODD for i in range(20): generated = self.integer_generator.generate() self.assertEqual(type(generated), int) self.assertGreaterEqual(generated, 1) self.assertLessEqual(generated, 100) self.assertEqual(generated % 2, 1) def test_generate_even_or_odd(self): self.integer_generator.even = IntegerGenerator.EVEN_OR_ODD even_found = False odd_found = False for i in range(20): generated = self.integer_generator.generate() if generated % 2 == 0: even_found = True if generated % 2 == 1: odd_found = True self.assertTrue(even_found) self.assertTrue(odd_found)
def setUp(self): self.integer_generator = IntegerGenerator()