unit test for flying team to string method

smash/web/models/flying_team.py

@@ -6,27 +6,6 @@ class FlyingTeam(models.Model):
     class Meta:
         app_label = 'web'
 
-    # doctor = models.ForeignKey(Worker, related_name='FlyingTeamDoctor',
-    #     verbose_name='Doctor'
-    # )
-    # nurse = models.ForeignKey(Worker, related_name='FlyingTeamNurse',
-    #     verbose_name='Nurse'
-    # )
-    # psychologist = models.ForeignKey(Worker, related_name='FlyingTeamPsychologist',
-    #     verbose_name='Psychologist'
-    # )
-    # datetime_called = models.DateTimeField(
-    #     verbose_name='Created on'
-    # )
-    # datetime_until = models.DateTimeField(
-    #     verbose_name='Disbanded on'
-    # )
-    #
-    # def __str__(self):
-    #     return "%s %s %s" % (self.doctor.last_name, self.nurse.last_name, self.psychologist.last_name)
-    #
-    # def __unicode__(self):
-    #     return "%s %s %s" % (self.doctor.last_name, self.nurse.last_name, self.psychologist.last_name)
     place = models.CharField(max_length=255, verbose_name='Place')
 
     def __str__(self):

smash/web/tests/models/test_flying_team.py

+import logging
+
+from django.test import TestCase
+
+from web.models import FlyingTeam
+
+logger = logging.getLogger(__name__)
+
+
+class FlyingTeamTests(TestCase):
+    def test_image_img(self):
+        flying_team = FlyingTeam(place="xxx")
+
+        self.assertTrue("x" in str(flying_team))