def test_reservations(self): for r in Reservation.get_reservations(self.user): self.assertIn(r, [self.reservation1, self.reservation2]) for r in self.user.get_reservations(): self.assertIn(r, [self.reservation1, self.reservation2]) self.assertEqual(Reservation.get_reservation_by_id(self.reservation1.id), \ self.reservation1) self.assertEqual(Reservation.get_reservation_by_id(self.reservation2.id), \ self.reservation2)