def test_populate():
zombies = 30
victims = 29
hunters = 33
map_x = 20
map_y = 20
testgame = Game(zombies, victims, hunters, 1, map_x, map_y)
testgame.populate()
zomb_count, vic_count, hunt_count = 0, 0, 0
tile_list = testgame.get_map().get_list()
blanks = 0
for list in tile_list:
for tile in list:
if tile.get_occupier() == "Zombie":
zomb_count+=1
elif tile.get_occupier() == "Victim":
vic_count+=1
elif tile.get_occupier() == "Hunter":
hunt_count+=1
else:
blanks+=1
assert_equal(zomb_count, zombies)
assert_equal(vic_count, victims)
assert_equal(hunt_count, hunters)
total_tiles = map_x*map_y
total_blanks = total_tiles - (zombies+victims+hunters)
assert_equal(blanks, total_blanks)
