def test_get_order_status(): """ Tests - get order status """ game = Game() game.clear_units() game.set_units('ITALY', 'A VEN') game.set_units('AUSTRIA', 'A VIE') game.set_orders('ITALY', 'A VEN - TYR') game.set_orders('AUSTRIA', 'A VIE - TYR') game.process() results = game.get_order_status() assert BOUNCE in results['ITALY']['A VEN'] assert BOUNCE in results['AUSTRIA']['A VIE'] assert BOUNCE in game.get_order_status(unit='A VEN') assert BOUNCE in game.get_order_status(unit='A VIE') assert BOUNCE in game.get_order_status('ITALY')['A VEN'] assert BOUNCE in game.get_order_status('AUSTRIA')['A VIE']
def test_clear_orders(): """ Test - Clear orders""" game = Game() game.clear_units() game.set_units('ITALY', 'A VEN') game.set_units('AUSTRIA', 'A VIE') game.set_orders('ITALY', 'A VEN - TYR') game.set_orders('AUSTRIA', 'A VIE - TYR') game.clear_orders() game.process() results = game.get_order_status() assert results['ITALY']['A VEN'] == [] assert results['AUSTRIA']['A VIE'] == []
def test_set_orders(): """ Test - Sets orders """ game = Game() game.clear_units() game.set_units('ITALY', 'A VEN') game.set_units('AUSTRIA', 'A VIE') game.set_units('FRANCE', 'A PAR') game.set_orders('ITALY', 'A VEN - TYR') game.set_orders('AUSTRIA', 'A VIE - TYR') game.set_orders('FRANCE', ['', '', 'A PAR - GAS', '', '', '']) game.set_orders('RUSSIA', '') game.set_orders('GERMANY', []) assert game.get_orders('FRANCE') == ['A PAR - GAS'] assert not game.get_orders('RUSSIA') assert not game.get_orders('GERMANY') game.process() results = game.get_order_status() assert BOUNCE in results['ITALY']['A VEN'] assert BOUNCE in results['AUSTRIA']['A VIE']