Use bakery if present

Closes #3
alsur · Jun 16, 2020 · ca3e92a · ca3e92a
1 parent 8baf77d
commit ca3e92a
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/admin_auto_tests/test_model.py b/admin_auto_tests/test_model.py
@@ -2,7 +2,10 @@
 
 import factory
 from django.contrib.auth import get_user_model
-from model_mommy import mommy
+try:
+    from model_baker import bakery as mommy
+except ImportError:
+    from model_mommy import mommy
 
 try:
     from django.core.urlresolvers import reverse