Exemplo n.º 1
0
def test_animal_serialize():
    # test DOG create/serialize
    a = Animal(AnimalType.CAT)
    actual = a.serialize()
    expected = '{"animaltype": 1}'
    assert actual == expected
    # test DOG create/serialize
    a = Animal(AnimalType.DOG)
    actual = a.serialize()
    expected = '{"animaltype": 2}'
    assert actual == expected
Exemplo n.º 2
0
def test_shelter_deq_pref_notfound():
    shelter = AnimalShelter()

    # Inc 10 (ALTERNATING)
    shelter.enqueue(Animal(AnimalType.CAT))
    shelter.enqueue(Animal(AnimalType.CAT))
    shelter.enqueue(Animal(AnimalType.CAT))

    actual = shelter.dequeue(AnimalType.DOG)
    expected = None
    assert actual == expected
Exemplo n.º 3
0
def test_shelter_deq_pref_found():
    shelter = AnimalShelter()

    # Inc 10 (ALTERNATING)
    shelter.enqueue(Animal(AnimalType.CAT))
    shelter.enqueue(Animal(AnimalType.CAT))
    shelter.enqueue(Animal(AnimalType.CAT))
    shelter.enqueue(Animal(AnimalType.DOG))
    shelter.enqueue(Animal(AnimalType.DOG))

    animal = shelter.dequeue(AnimalType.DOG)
    actual = animal.serialize()
    expected = '{"animaltype": 2}'
    assert actual == expected
Exemplo n.º 4
0
def helper_shelter_enq10():
    shelter = AnimalShelter()

    # Inc 10 (ALTERNATING)
    shelter.enqueue(Animal(AnimalType.CAT))
    shelter.enqueue(Animal(AnimalType.DOG))
    shelter.enqueue(Animal(AnimalType.CAT))
    shelter.enqueue(Animal(AnimalType.DOG))
    shelter.enqueue(Animal(AnimalType.CAT))
    shelter.enqueue(Animal(AnimalType.DOG))
    shelter.enqueue(Animal(AnimalType.CAT))
    shelter.enqueue(Animal(AnimalType.DOG))
    shelter.enqueue(Animal(AnimalType.CAT))
    shelter.enqueue(Animal(AnimalType.DOG))

    return shelter
Exemplo n.º 5
0
def test_shelter_enq_deq():
    shelter = AnimalShelter()
    shelter.enqueue(Animal(AnimalType.CAT))

    animal = shelter.dequeue()
    expected = '{"animaltype": 1}'
    actual = animal.serialize()
    assert actual == expected
Exemplo n.º 6
0
def test_class_exists():
    AnimalShelter()
    Animal(AnimalType.CAT)
    Cat()
    Dog()
def test_animal():
    animal = Animal('cat')
    assert animal
def test_enqueue_bad_value():
    with pytest.raises(TypeError):
        Animal('lizard')
def test_animal_caps():
    caps = Animal('DOG')
    assert caps
def test_animal_capital_letter():
    cat = Animal('CAT')
    assert cat