def test_def_validate_roll_fail(roll, expected_keepers): prints = ['No way pal', roll, 'No way pal',roll] inputs = ['0','0', str(expected_keepers[0])] def my_print(msg, *args): assert msg == prints.pop(0) def my_input(msg, *args): assert msg == 'Enter dice to keep: ' return inputs.pop(0) game = Game(my_print, my_input) keepers = game.validate_roll(roll) assert keepers == expected_keepers
def test_def_validate_roll_success(roll, expected_keepers): def my_print(msg, *args): assert msg == roll def my_input(msg, *args): assert msg == 'Enter dice to keep: ' keeper_string = '' for val in expected_keepers: keeper_string += str(val) return keeper_string game = Game(my_print, my_input) keepers = game.validate_roll(roll) assert keepers == expected_keepers