def test_shape(): shape = (16, 16) wator = WaTor(shape=shape, nfish=16, nsharks=4) print(wator.creatures) assert wator.creatures.shape == shape assert wator.count_fish() == 16 assert wator.count_sharks() == 4 assert ((wator.creatures >= -10) & (wator.creatures <= 5)).all()
def test_creatures(): creatures = numpy.zeros((8, 8)) creatures[2, 4] = 3 creatures[1, :] = -5 wator = WaTor(creatures) print(wator.creatures) assert (creatures == wator.creatures).all() assert wator.count_fish() == 1 assert wator.count_sharks() == 8
def test_shape_full_of_something(): shape = (16, 16) total = 16 * 16 nfish = total // 5 nsharks = total - nfish wator = WaTor(shape=shape, nfish=nfish, nsharks=nsharks) print(wator.creatures) assert wator.count_fish() == nfish assert wator.count_sharks() == nsharks
def test_shape_custom_age(): shape = (16, 16) age_fish = 2 age_shark = 20 wator = WaTor(shape=shape, nfish=16, nsharks=4, age_fish=age_fish, age_shark=age_shark) print(wator.creatures) assert wator.creatures.shape == shape assert wator.count_fish() == 16 assert wator.count_sharks() == 4 assert ((wator.creatures >= -age_shark) & (wator.creatures <= age_fish)).all()
def test_random_geenerator_speed(): for i in range(20): wator = WaTor(shape=SHAPE, nfish=N, nsharks=N) print(i) assert wator.count_fish() == N assert wator.count_sharks() == N
def test_shape_full_of_shark(): shape = (16, 16) wator = WaTor(shape=shape, nfish=0, nsharks=16 * 16) print(wator.creatures) assert wator.count_fish() == 0 assert wator.count_sharks() == 16 * 16