def test_treatment_unicode(self): '''This tests the unicode representation of a :class:`~mousedb.data.models.Treatment`.''' test_treatment = Treatment(treatment = 'Test Treatment', diet = Diet.objects.get(pk=1), environment = Environment.objects.get(pk=1)) test_treatment.save() test_treatment.animals.add(Animal.objects.get(pk=1)) test_treatment.researchers.add(Researcher.objects.get(pk=1)) self.assertEqual(test_treatment.__unicode__(), "Test Treatment")