class PetTests(TestCase):
def setUp(self):
self.pet = Pet(name="test_cat", species="Cat", age=2)
def tearDown(self):
Pet.query.delete()
db.session.commit()
def test_image(self):
self.assertEqual(
self.pet.set_image(),
'https://cdn3.f-cdn.com/ppic/3151051/logo/7139949/cat%20fb.jpg')
