def test_get_total_hours_from_workitems(self): user = UserFactory(username="******") WorkItemFactory.create_batch(4, user=user, hours=2) items = WorkItem.objects.filter(user=user) self.assertEqual(get_total_hours_from_workitems(items), 8.0)