def test_enqueue_cat():
a = AnimalShelter()
c = Cat()
a.enqueue(c)
actual = a.dequeue("cat")
excepted = c
assert actual == excepted
def test_animal_enqueue_one():
animal_instance = Animal("cat", "Fluffers")
shelter_instance = AnimalShelter()
shelter_instance.enqueue(animal_instance)
actual = shelter_instance.cats.peek()
expected = animal_instance
assert actual == expected
def test_animal_dequeue_one():
animal_instance = Animal("cat", "Carl")
shelter_instance = AnimalShelter()
shelter_instance.enqueue(animal_instance)
actual = shelter_instance.dequeue("cat")
expected = animal_instance
assert actual == expected
def test_dog_enqueue():
expected = "Bobi"
dog1 = Dog('Bobi')
animal_sh = AnimalShelter()
animal_sh.enqueue(dog1)
actual = f"{animal_sh.dog}"
assert actual == expected
def test_cat_enqueue():
expected = "kitty"
cat1 = Cat('kitty')
animal_sh = AnimalShelter()
animal_sh.enqueue(cat1)
actual = f"{animal_sh.cat}"
assert actual == expected
def test_single_animal():
shelter = AnimalShelter()
cat = Cat()
shelter.enqueue(cat)
actual = shelter.dequeue("cat")
expected = cat
assert actual == expected
def test_shelter_enqueue_dog():
shelter = AnimalShelter()
shelter.enqueue(jora)
actual_1 = shelter.dog.front.value.name
actual_2 = shelter.dog.rear.value.name
expected = 'Jora'
assert actual_1 == expected
assert actual_2 == expected
def test_shelter_enqueue_cat():
shelter = AnimalShelter()
shelter.enqueue(ferran)
actual_1 = shelter.cat.front.value.name
actual_2 = shelter.cat.rear.value.name
expected = 'Ferran'
assert actual_1 == expected
assert actual_2 == expected
def test_animal_enqueue_several():
animal_instance_one = Animal("dog", "Doggy")
animal_instance_two = Animal("cat", "Kitty")
shelter_instance = AnimalShelter()
shelter_instance.enqueue(animal_instance_one)
shelter_instance.enqueue(animal_instance_two)
actual = shelter_instance.cats.peek()
expected = animal_instance_two
assert actual == expected
def test_multi_dequeue_cat():
shelter = AnimalShelter()
shelter.enqueue(ferran)
shelter.enqueue(jora)
shelter.enqueue(raya)
shelter.enqueue(athena)
shelter.dequeue('cat')
shelter.dequeue('cat')
actual = shelter.cat.front
expected = None
assert actual == expected
def test_cat_wait_after_multi_dequeue_dog():
shelter = AnimalShelter()
shelter.enqueue(ferran)
shelter.enqueue(jora)
shelter.enqueue(raya)
shelter.enqueue(athena)
shelter.dequeue('dog')
shelter.dequeue('dog')
actual = shelter.cat_wait
expected = 2
assert actual == expected
def test_dog_wait_after_dequeue_cat():
shelter = AnimalShelter()
shelter.enqueue(ferran)
shelter.enqueue(jora)
shelter.dequeue('cat')
actual = shelter.dog_wait
expected = 1
assert actual == expected
def sample_shelter():
shelter = AnimalShelter()
shelter.enqueue('cat')
shelter.enqueue('cat')
shelter.enqueue('dog')
shelter.enqueue('cat')
shelter.enqueue('dog')
return shelter
def test_dequeue_pref_cat():
shelter = AnimalShelter()
shelter.enqueue(ferran)
shelter.enqueue(jora)
actual = shelter.dequeue('cat')
expected = 'Ferran'
assert actual == expected
def test_dequeue_pref_dog():
shelter = AnimalShelter()
shelter.enqueue(ferran)
shelter.enqueue(jora)
actual = shelter.dequeue('dog')
expected = 'Jora'
assert actual == expected
def test_dequq_without_param():
shelter = AnimalShelter()
shelter.enqueue('cat')
shelter.enqueue('dog')
shelter.enqueue('cat')
shelter.enqueue('dog')
shelter.dequeue()
actual = str(shelter)
expected = 'dog -> cat -> dog -> Null'
assert actual == expected
def test_animal_dequeue_several_types():
animal_instance_one = Animal("cat", "Meow")
animal_instance_two = Animal("dog", "Bark")
shelter_instance = AnimalShelter()
shelter_instance.enqueue(animal_instance_one)
shelter_instance.enqueue(animal_instance_two)
shelter_instance.dequeue("cat")
actual = shelter_instance.dequeue("dog")
expected = animal_instance_two
assert actual == expected
def test_animal_dequeue_several():
animal_instance_one = Animal("dog", "Trump")
animal_instance_two = Animal("dog", "Ivanka")
shelter_instance = AnimalShelter()
shelter_instance.enqueue(animal_instance_one)
shelter_instance.enqueue(animal_instance_two)
shelter_instance.dequeue("dog")
actual = shelter_instance.dequeue("dog")
expected = animal_instance_two
assert actual == expected
def test_animal_dequeue():
shelter = AnimalShelter()
cat = Cat()
dog = Dog()
shelter.enqueue(cat)
shelter.enqueue(dog)
shelter.enqueue(dog)
actual = shelter.dequeue('dog')
expected = dog
assert actual == expected
def test_enqueue():
shelter = AnimalShelter()
shelter.enqueue('cat')
shelter.enqueue('dog')
shelter.enqueue('cat')
actual = str(shelter)
expected = 'cat -> dog -> cat -> Null'
assert actual == expected
def test_shelter_enqueue_multi_cat():
shelter = AnimalShelter()
shelter.enqueue(ferran)
shelter.enqueue(raya)
actual_1 = shelter.cat.front.value.name
expected_1 = 'Ferran'
assert actual_1 == expected_1
actual_2 = shelter.cat.rear.value.name
expected_2 = 'Raya'
assert actual_2 == expected_2
def test_shelter_enqueue_multi_dog():
shelter = AnimalShelter()
shelter.enqueue(jora)
shelter.enqueue(athena)
actual_1 = shelter.dog.front.value.name
expected_1 = 'Jora'
assert actual_1 == expected_1
actual_2 = shelter.dog.rear.value.name
expected_2 = 'Athena'
assert actual_2 == expected_2
def queue_dog_test():
dog1 = Dog('bobi')
dog2 = Dog('sasi')
dog3 = Dog('soso')
dog4 = Dog('lulu')
dogs = AnimalShelter()
dogs.enqueue(dog1)
dogs.enqueue(dog2)
dogs.enqueue(dog3)
dogs.enqueue(dog4)
return dogs.dog
def queue_cat_test():
cat1 = Cat('kitty')
cat2 = Cat('lucy')
cat3 = Cat('sabi')
cat4 = Cat('radi')
cats = AnimalShelter()
cats.enqueue(cat1)
cats.enqueue(cat2)
cats.enqueue(cat3)
cats.enqueue(cat4)
return cats.cat
def test_enqueue_cat():
ams = AnimalShelter()
ams.enqueue('cat')
assert ams.cat_queue.peek() == 'cat'
def test_dequeue_not_cat_or_dog():
ams = AnimalShelter()
with pytest.raises(InvalidOperationError):
ams.dequeue('mouse')
def test_dequeue_dog():
ams = AnimalShelter()
ams.enqueue('cat')
ams.enqueue('dog')
ams.dequeue('dog')
assert ams.dog_queue.is_empty() == True
def test_enqueue_dog():
ams = AnimalShelter()
ams.enqueue('dog')
assert ams.dog_queue.peek() == 'dog'
def test_AnimalShelter__init__():
test_shelter = AnimalShelter()
actual = test_shelter.shelter
expected = []
assert actual == expected
def test_animal_enqueue_invalid_type():
animal_instance = Animal("cow", "Moo")
shelter_instance = AnimalShelter()
with pytest.raises(Exception):
shelter.enqueue(animal_instance)
