def test_solve_success(): params = BoxFactory.attributes() box = Box(**params) box._levels[0].solve = Mock(return_value = True) assert box.solve() == True assert box.current_level() == 2
def test_solve_failure(): params = BoxFactory.attributes() box = Box(**params) box._levels[0].solve = Mock(return_value = False) assert box.solve() == False assert box.current_level() == 1
def test_current_level(): params = BoxFactory.attributes() box = Box(**params) assert box.current_level() == 1