From 6ca53678699a4eececa3b178f0f7e051956bf580 Mon Sep 17 00:00:00 2001
From: Andrew Godwin <andrew@aeracode.org>
Date: Wed, 30 Apr 2014 15:11:29 -0700
Subject: [1.7.x] Cleanup failing test w/contenttypes and remove useless graph
 load

---
 tests/migrations/test_state.py | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

(limited to 'tests')

diff --git a/tests/migrations/test_state.py b/tests/migrations/test_state.py
index e5cee5f6cd..afd2f8b8b9 100644
--- a/tests/migrations/test_state.py
+++ b/tests/migrations/test_state.py
@@ -361,7 +361,10 @@ class StateTests(TestCase):
         project_state = ProjectState(real_apps=["contenttypes"])
         project_state.add_model_state(ModelState.from_model(TestModel))
         rendered_state = project_state.render()
-        self.assertEqual(len(rendered_state.get_models()), 2)
+        self.assertEqual(
+            len([x for x in rendered_state.get_models() if x._meta.app_label == "migrations"]),
+            1,
+        )
 
 
 class ModelStateTests(TestCase):
-- 
cgit v1.3