Пример #1
0
def test_shelter_enqueue():
    shelter = AnimalShelter()

    shelter.enqueue(Cat("1"))
    assert shelter.queue1.front.value == "1"
    assert shelter.queue1.rear.value == "1"
    shelter.enqueue(Dog("2"))
    assert shelter.queue1.rear.value == "2"
    shelter.enqueue(Cat("3"))
    assert shelter.queue1.rear.value == "3"
    shelter.enqueue(Dog("4"))
    assert shelter.queue1.rear.value == "4"
    assert shelter.queue1.front.value == "1"
Пример #2
0
def test_enqueu_cat_dog():
    animals = AnimalShelter()
    alex = Dog('alex')
    aln = Dog('Aln')
    Sherry = Cat('Sherry')
    Semsem = Cat('Semsem')
    animals.enqueue(alex)
    animals.enqueue(aln)
    animals.enqueue(Sherry)
    animals.enqueue(Semsem)

    expected = ('alex', 'Aln', 'Sherry', 'Semsem')
    actual = alex.name, aln.name, Sherry.name, Semsem.name
    assert expected == actual
Пример #3
0
def test_dequeue_cat():
    animals = AnimalShelter()
    alex = Dog('alex')
    aln = Dog('Aln')
    Sherry = Cat('Sherry')
    Semsem = Cat('Semsem')
    animals.enqueue(alex)
    animals.enqueue(aln)
    animals.enqueue(Sherry)
    animals.enqueue(Semsem)

    expected = 'Sherry'
    actual = animals.dequeue('cat')
    assert expected == actual
Пример #4
0
def fixed_shelter():
    shelter = AnimalShelter()
    shelter.enqueue(Cat('Tom'))
    shelter.enqueue(Dog('Spike'))
    shelter.enqueue(Cat('Mdhat'))

    return shelter
def test_dequeue_and_enqueue():
    mylo = Dog('mylo')
    reixy = Dog('mylo')
    luna = Dog('luna')
    kitty = Cat('kitty')
    lily = Cat('lily')
    a = AnimalShelter()
    a.enqueue(mylo)
    a.enqueue(reixy)
    a.enqueue(luna)
    a.enqueue(kitty)
    a.enqueue(lily)

    assert a.dequeue('cat') == 'kitty'
    assert a.dequeue('cat') == 'lily'
    assert a.dequeue('dog') == 'mylo'
    assert a.dequeue('dog') == 'mylo'
    assert a.dequeue('dog') == 'luna'
Пример #6
0
def test_dequeue_any():
    animals = AnimalShelter()
    aln = Dog('Aln')
    Sherry = Cat('Sherry')
    animals.enqueue(aln)
    animals.enqueue(Sherry)

    expected = None
    actual = animals.dequeue('any')
    assert expected == actual
Пример #7
0
def test_shelter_dequeue():
    shelter = AnimalShelter()

    shelter.enqueue(Cat("1"))
    shelter.enqueue(Dog("2"))
    shelter.enqueue(Cat("3"))
    shelter.enqueue(Cat("a"))
    shelter.enqueue(Dog("4"))

    pet_adopted = shelter.dequeue('cat')
    assert pet_adopted.type == 'cat'
    assert pet_adopted.value == '1'

    pet_adopted = shelter.dequeue('cat')
    assert pet_adopted.type == 'cat'
    assert pet_adopted.value == '3'

    pet_adopted = shelter.dequeue('dog')
    assert shelter.queue1.front.value == "a"
    assert pet_adopted.type == 'dog'
    assert pet_adopted.value == '2'
Пример #8
0
def test_Animal_Shelter(fixed_shelter):
    fixed_shelter.enqueue(Dog('JOJO'))
    assert str(
        fixed_shelter) == f'{{Tom}}->{{Spike}}->{{Mdhat}}->{{JOJO}}->NULL'

    assert fixed_shelter.dequeue('Dog') == 'Spike'
    assert str(fixed_shelter) == f'{{Tom}}->{{Mdhat}}->{{JOJO}}->NULL'

    assert fixed_shelter.dequeue('Bird') == 'NULL'
    assert str(fixed_shelter) == f'{{Tom}}->{{Mdhat}}->{{JOJO}}->NULL'

    assert fixed_shelter.dequeue('Cat') == 'Tom'
    assert str(fixed_shelter) == f'{{Mdhat}}->{{JOJO}}->NULL'

    assert fixed_shelter.dequeue() == 'Mdhat'
    assert str(fixed_shelter) == f'{{JOJO}}->NULL'
def test_creating_dog_object():
    assert Dog().name == 'Ima dog'
Пример #10
0
def test_shelter_dequeue_capital():
    shelter = AnimalShelter()
    shelter.enqueue(Cat("1"))
    shelter.enqueue(Dog("2"))
    pet_adopted = shelter.dequeue('Dog')
    assert pet_adopted.value == '2'
Пример #11
0
def test_shelter_dequeue_not_in_list2():
    shelter = AnimalShelter()
    shelter.enqueue(Cat("1"))
    shelter.enqueue(Dog("2"))
    pet_adopted = shelter.dequeue('parrot')
    assert pet_adopted == None
Пример #12
0
def test_dog():
    dog = Dog("Sam")
    assert dog.value == 'Sam'
    assert dog.type == 'dog'
    assert dog.next == None