def test_change_battalion_strength(self): horses, elephants, armoured_tanks, sling_guns = 100, 50, 10, 5 army = Army(horses=horses, elephants=elephants, armoured_tanks=armoured_tanks, sling_guns=sling_guns) army.update_battalion_strength(BattalionType.HORSES, -50) army.update_battalion_strength(BattalionType.SLINGGUNS, 5) assert army.get_battalion_strength(BattalionType.HORSES) == 50 assert army.get_battalion_strength(BattalionType.SLINGGUNS) == 10
def test_change_battalion_strength_1(self): army = Army() army.update_battalion_strength(BattalionType.HORSES, 50) assert army.get_battalion_strength(BattalionType.HORSES) == 50