def test_draw_question_cards(self): state = GameState() state.interrogation_deck = [(8, 'H'), (3, '$'), (1, '$'), (1, 'L')] draw_question_cards(state) self.assertItemsEqual([(1, 'L'), (1, '$'), (3, '$')], state.question_cards) self.assertEqual([(8, 'H')], state.interrogation_deck)
def test_draw_question_cards_when_there_are_not_enough_cards_in_the_interrogation_deck(self): state = GameState() state.interrogation_deck = [(4, 'L')] state.discard_deck = [(3, '$'), (6, 'H'), (7, 'L')] draw_question_cards(state) self.assertEqual(3, len(state.question_cards)) self.assertIn((4, 'L'), state.question_cards) self.assertEqual(0, len(state.discard_deck)) self.assertEqual(1, len(state.interrogation_deck))