Brought tests in line with mptt branch. Added assertQueryLimit method as a rough...

[philo.git] / models / fields.py

diff --git a/models/fields.py b/models/fields.py
index 85c5583..3e43c0f 100644 (file)
--- a/models/fields.py
+++ b/models/fields.py
@@ -92,7 +92,7 @@ class ForeignKeyAttributeDescriptor(AttributeFieldDescriptor):
 
 class ManyToManyAttributeDescriptor(AttributeFieldDescriptor):
        def __set__(self, instance, value):
 
 class ManyToManyAttributeDescriptor(AttributeFieldDescriptor):
        def __set__(self, instance, value):

-               if isinstance(value, models.QuerySet):
+               if isinstance(value, models.query.QuerySet):

                        if self.field in instance._removed_attribute_registry:
                                instance._removed_attribute_registry.remove(self.field)
                        instance._added_attribute_registry[self.field] = value
                        if self.field in instance._removed_attribute_registry:
                                instance._removed_attribute_registry.remove(self.field)
                        instance._added_attribute_registry[self.field] = value