def test_create(self): current_count = Room.all().count() Room.create("New Room", 'living_space') self.assertEqual(Room.all().count(), current_count + 1)
def setUp(self): Base.metadata.create_all(engine) self.room = Room.create("Room Name", "office")