def test_disconnect(self):
""" Test Disconnet """
Pet.disconnect()
pet = Pet("fido", "dog", False)
self.assertRaises(AttributeError, pet.save)
def test_connect(self):
"""Test Connect"""
Pet.disconnect()
Pet.connect()
pet = Pet("fido", "dog", False)
pet.create()
