def test_get_category_name_not_all(self): user = User() user.created = timezone.now().strftime('%Y-%m-%d %H:%M:%S') user.save() category = Category() category.name = 'Test category' category.user = user category.save() self.assertEqual( Category.get_category_name(category.id), 'Test category' )
def test_get_category_name_all(self): self.assertEqual(Category.get_category_name(0), 'All')