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()