from pet import Pet
import unittest

class MyUnitTestClass(unittest.TestCase):
        
    def test_1(self):
        pet1 = Pet("Sparky", "Dog")
        self.assertEqual(str(pet1), \
'''Name: Sparky
Animal Type: Dog
Is Vaccinated: No
''')
        self.assertEqual(pet1.name, "Sparky")
        self.assertEqual(pet1.animal_type, "Dog")
        self.assertEqual(pet1.is_vaccinated( ), "No")
        pet1.vaccinate( )
        self.assertEqual(pet1.is_vaccinated( ), "Yes")
        
if __name__ == '__main__':
    unittest.main()