summaryrefslogtreecommitdiff
path: root/factory/base.py
diff options
context:
space:
mode:
authorRaphaël Barrois <raphael.barrois@polytechnique.org>2012-11-14 23:53:26 +0100
committerRaphaël Barrois <raphael.barrois@polytechnique.org>2012-11-15 00:03:37 +0100
commita5184785b1229e02ec75506db675a2377b32c297 (patch)
tree415a1f83a97d6d128c1e5c75cba8a228608e5699 /factory/base.py
parent048fb1cc935a2ccbc5ca7af81c9390e218a1080b (diff)
downloadfactory-boy-a5184785b1229e02ec75506db675a2377b32c297.tar
factory-boy-a5184785b1229e02ec75506db675a2377b32c297.tar.gz
Keep FACTORY_FOR around.
And add a test too. Signed-off-by: Raphaël Barrois <raphael.barrois@polytechnique.org>
Diffstat (limited to 'factory/base.py')
-rw-r--r--factory/base.py3
1 files changed, 0 insertions, 3 deletions
diff --git a/factory/base.py b/factory/base.py
index 3d4e8c1..2b78e2b 100644
--- a/factory/base.py
+++ b/factory/base.py
@@ -236,9 +236,6 @@ class FactoryMetaClass(BaseFactoryMetaClass):
associated_class = cls._discover_associated_class(class_name, attrs,
inherited_associated_class)
- # Remove the FACTORY_CLASS_DECLARATION attribute from attrs, if present.
- attrs.pop(FACTORY_CLASS_DECLARATION, None)
-
# If inheriting the factory from a parent, keep a link to it.
# This allows to use the sequence counters from the parents.
if associated_class == inherited_associated_class: