def test_get_key_pegs(self):
self.assertEqual(Mastermindv2.get_key_pegs(1, 2), ['Red', 'Red', 'Empty', 'Empty'])
self.assertEqual(Mastermindv2.get_key_pegs(3, 4), ['Red', 'Red', 'Red', 'Red'])
self.assertEqual(Mastermindv2.get_key_pegs(4, 3), ['Red', 'Red', 'Red', 'White'])
def test_check_order(self):
self.assertEqual(Mastermindv2.check_order(['1', '1', '2', '4'], ['1', '1', '2', '4']), 4)
self.assertEqual(Mastermindv2.check_order(['1', '1', '2', '4'], ['2', '1', '3', '4']), 2)
self.assertEqual(Mastermindv2.check_order(['2', '1', '2', '1'], ['3', '1', '4', '3']), 1)
def test_check_numbers(self):
self.assertEqual(Mastermindv2.check_numbers(['1', '1', '2', '4'], ['1', '1', '2', '4']), 4)
self.assertEqual(Mastermindv2.check_numbers(['1', '1', '2', '4'], ['2', '1', '3', '4']), 3)
self.assertEqual(Mastermindv2.check_numbers(['2', '1', '2', '1'], ['3', '1', '4', '1']), 2)
def test_get_code_breaker_attempt(self):
self.assertEqual(Mastermindv2.get_code_breaker_attempt(), Mastermindv2.generate_code())
def test_wrong_characters(self):
self.assertEqual(Mastermindv2.wrong_characters(['1', '2', '4', '3']), False)
self.assertEqual(Mastermindv2.wrong_characters(['b', 'e', 'e', 'p']), True)
self.assertEqual(Mastermindv2.wrong_characters(['1','a', 'b', 'c']), True)
def test_wrong_code_length(self):
self.assertEqual(Mastermindv2.wrong_code_length(['1', '2', '4', '3']), False)
self.assertEqual(Mastermindv2.wrong_code_length(['1', '2', '4', '5', '4']), True)
self.assertEqual(Mastermindv2.wrong_code_length(['1', '2']), True)