def test_template_tag_position_in_queue(self): gpu = self.devices[0].gpus.first() other = mommy.make(User) mommy.make(Reservation, gpu=gpu, user=other) mommy.make(Reservation, gpu=gpu, user=self.user) self.assertEqual(queue_position(gpu, self.user), 1)
def test_template_tag_position_in_queue_not_reserved(self): gpu = self.devices[0].gpus.first() self.assertIsNone(queue_position(gpu, self.user))