Add unit tests for enrich function

References #11

src/bio2bel_mirtarbase/enrich.py

@@ -0,0 +1,12 @@
+# -*- coding: utf-8 -*-
+
+from pybel_tools import pipeline
+
+
+@pipeline.in_place_mutator
+def enrich_proteins(graph):
+    """Adds all of the miRNA inhibitors of the proteins in the graph
+
+    :param pybel.BELGraph graph: A BEL graph
+    """
+    raise NotImplemented

tests/test_enrich.py

@@ -0,0 +1,23 @@
+# -*- coding: utf-8 -*-
+
+import unittest
+
+from bio2bel_mirtarbase.enrich import enrich_proteins
+from pybel import BELGraph
+from pybel.constants import PROTEIN, MIRNA
+
+c = PROTEIN, 'HGNC', 'HIF1A'
+t1 = MIRNA, 'MTB', 'MIRT000002'
+
+
+class TestEnrich(unittest.TestCase):
+    def test_enrich(self):
+        g = BELGraph()
+
+        g.add_simple_node(*c)
+
+        self.assertEqual(1, g.number_of_nodes())
+
+        enrich_proteins(g)
+
+        self.assertTrue(g.has_edge(t1, c))