visualize: convert to pytest

Author: jelly

conftest.py

@@ -0,0 +1,20 @@
+import pytest
+
+from django.core.management import call_command
+
+
+
+@pytest.fixture
+def arches(db):
+    call_command('loaddata', 'main/fixtures/arches.json')
+
+
+@pytest.fixture
+def repos(db):
+    call_command('loaddata', 'main/fixtures/repos.json')
+
+
+@pytest.fixture
+def package(db):
+    # TODO(jelle): create own parameter based version
+    call_command('loaddata', 'main/fixtures/package.json')

visualize/tests.py

@@ -1,11 +1,4 @@
-from django.test import TestCase
-
-
-class VisualeTest(TestCase):
-    fixtures = ['main/fixtures/arches.json', 'main/fixtures/repos.json',
-                'main/fixtures/package.json']
-
-    def test_urls(self):
-        for url in ['', 'by_repo/', 'by_arch/']:
-            response = self.client.get('/visualize/{}'.format(url))
-            self.assertEqual(response.status_code, 200)
+def test_urls(client, arches, repos, package):
+    for url in ['', 'by_repo/', 'by_arch/']:
+        response = client.get('/visualize/{}'.format(url))
+        assert response.status_code == 200