def test_adding_kitty_participants_to_race(self):
kitty_one = nya.Cat("desperado")
kitty_two = nya.Cat("el niño")
kitty_three = nya.Cat("catnip")
kitty_four = nya.Cat("desperado")
race = nya.NyaRace(500)
race.add_kitty_participant(kitty_one)
race.add_kitty_participant(kitty_two)
race.add_kitty_participant(kitty_three)
race.add_kitty_participant(kitty_four)
self.assertEqual(4, len(race._kitty_participants))
# Can't add kitties that are already racing
kitty_racing = nya.Cat("already racing")
kitty_racing.is_racing = True
race.add_kitty_participant(kitty_racing)
self.assertEqual(4, len(race._kitty_participants))
def test_cat_race(self):
kitty_one = nya.Cat("desperado")
kitty_one.food = 750
kitty_one.agility = 5
kitty_two = nya.Cat("el niño")
kitty_two.food = 500
kitty_two.agility = 10
kitty_three = nya.Cat("catnip")
kitty_three.food = 800
kitty_three.agility = 4
kitty_four = nya.Cat("hotail")
kitty_four.food = 1000
kitty_four.agility = 3
race = nya.NyaRace(100)
race.add_kitty_participant(kitty_one)
race.add_kitty_participant(kitty_two)
race.add_kitty_participant(kitty_three)
race.add_kitty_participant(kitty_four)
expected_results = (
[kitty_four, kitty_three, kitty_one, kitty_two],
[kitty_four, kitty_three, kitty_one, kitty_two],
[kitty_four, kitty_three, kitty_one, kitty_two],
[kitty_four, kitty_two, kitty_three, kitty_one],
[kitty_four, kitty_two, kitty_three, kitty_one],
[kitty_two, kitty_four, kitty_one, kitty_three],
[kitty_two, kitty_one, kitty_three, kitty_four],
[kitty_two, kitty_one, kitty_three, kitty_four],
[kitty_two, kitty_one, kitty_three, kitty_four],
[kitty_two, kitty_one, kitty_three, kitty_four],
)
received_results = tuple(race.step())
print(expected_results)
print(received_results)
self.assertEqual(expected_results, received_results)
def test_cat_race(self):
kitty_one = nya.Cat("desperado")
kitty_one.food = 750
kitty_one.agility = 5
kitty_two = nya.Cat("el niño")
kitty_two.food = 500
kitty_two.agility = 10
kitty_three = nya.Cat("catnip")
kitty_three.food = 800
kitty_three.agility = 4
kitty_four = nya.Cat("hotail")
kitty_four.food = 1000
kitty_four.agility = 3
race = nya.NyaRace(100)
race.add_kitty_participant(kitty_one)
race.add_kitty_participant(kitty_two)
race.add_kitty_participant(kitty_three)
race.add_kitty_participant(kitty_four)
expected_results = (
[
kitty_four,
kitty_three,
kitty_one,
kitty_two,
],
[
kitty_four,
kitty_three,
kitty_one,
kitty_two,
],
[
kitty_four,
kitty_three,
kitty_one,
kitty_two,
],
[
kitty_four,
kitty_two,
kitty_three,
kitty_one,
],
[
kitty_four,
kitty_two,
kitty_three,
kitty_one,
],
[
kitty_two,
kitty_four,
kitty_one,
kitty_three,
],
[
kitty_two,
kitty_one,
kitty_three,
kitty_four,
],
[
kitty_two,
kitty_one,
kitty_three,
kitty_four,
],
[
kitty_two,
kitty_one,
kitty_three,
kitty_four,
],
[
kitty_two,
kitty_one,
kitty_three,
kitty_four,
],
)
for expected_state, race_state in zip(expected_results, race.step()):
self.assertEqual(expected_state, race_state)
def test_race_participant_type(self):
# race _kitty_participants should not be altered from it's original type.
race_test = nya.NyaRace(500)
self.assertEqual(type(race_test._kitty_participants), set)