diff --git a/models.py b/models.py index b1a7dcc..f01f2fc 100644 --- a/models.py +++ b/models.py @@ -12,6 +12,9 @@ class EavAttributeLabel(models.Model): name = models.CharField(_(u"name"), db_index=True, unique=True, max_length=100) + def __unicode__(self): + return self.name + class EavAttribute(models.Model): ''' @@ -68,7 +71,7 @@ class EavAttribute(models.Model): super(EavAttribute, self).save(*args, **kwargs) def add_label(self, label): - pass + self.labels.get_or_create(name=label) def get_value_for_entity(self, entity): @@ -137,7 +140,6 @@ class EavValue(models.Model): value = property(_get_value, _set_value) - def __unicode__(self): return u"%s - %s: \"%s\"" % (self.object, self.attribute.name, self.value) diff --git a/tests/tests.py b/tests/tests.py index 0d0275e..bf2439a 100644 --- a/tests/tests.py +++ b/tests/tests.py @@ -24,7 +24,6 @@ class EavBasicTests(TestCase): attribute=self.attribute, value_text='Denver') - def tearDown(self): EavRegistry.unregister(Patient)