Corrected import and began testing NetworkX graph exports

Author: JackAtOmenApps

django_postgresql_dag/models.py

@@ -18,7 +18,7 @@
 from django.core.exceptions import ValidationError
 
 from .exceptions import NodeNotReachableException
-from .transformations import *
+from .transformations import _filter_order
 
 LIMITING_FK_EDGES_CLAUSE_1 = (
     """AND second.{fk_field_name}_{pk_name} = %(limiting_fk_edges_instance_pk)s"""

setup.py

@@ -3,7 +3,7 @@
 import os
 from setuptools import setup
 
-version = '0.0.19'
+version = '0.0.20'
 
 classifiers = [
     "Development Status :: 3 - Alpha",

tests/test.py

@@ -4,7 +4,8 @@
 
 from django.test import TestCase
 from django.core.exceptions import ValidationError
-from django_postgresql_dag.models import NodeNotReachableException
+from django_postgresql_dag.models import NodeNotReachableException, GraphModelsCannotBeParsedException, IncorrectUsageException
+from django_postgresql_dag.transformations import _filter_order, edges_from_nodes_queryset, nodes_from_edges_queryset, nx_from_queryset, model_to_dict
 
 # from .dag_output import expected_dag_output
 from .models import NetworkNode, NetworkEdge
@@ -226,6 +227,10 @@ def test_02_dag(self):
         log.debug("ancestors")
         self.assertEqual([p.name for p in c1.ancestors()], ["root", "a3", "b4"])
         self.assertFalse(c1.is_island())
+        
+        # Test is we can properly export to a NetworkX graph
+        nx_out = nx_from_queryset(c1.ancestors_and_self(), graph_attributes_dict={"test": "test"}, node_attribute_fields_list=["id", "name"], edge_attribute_fields_list=["id", "name"])
+        self.assertEqual(nx_out.graph, {"test": "test"})
 
         """
         Simulate a basic irrigation canal network