Python Army.update_battalion_strength Examples

Example #1

File: test_army.py Project: benneyman/War

 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

Example #2

File: test_army.py Project: benneyman/War

 def test_change_battalion_strength_1(self):
     army = Army()
     army.update_battalion_strength(BattalionType.HORSES, 50)
     assert army.get_battalion_strength(BattalionType.HORSES) == 50