Fix broken imports and Python path.

cfe-lab · Jan 25, 2018 · 774bb04 · 774bb04
1 parent a61ed59
commit 774bb04
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 7 deletions.
diff --git a/.idea/pyvdrm.iml b/.idea/pyvdrm.iml
diff --git a/pyvdrm/tests/test_vcf.py b/pyvdrm/tests/test_vcf.py
@@ -1,6 +1,5 @@
 import unittest
 from pyvdrm.vcf import Mutation, MutationSet, VariantCalls
-from vcf import VariantCalls
 
 
 class TestMutation(unittest.TestCase):
@@ -428,10 +427,6 @@ def test_immutable(self):
             calls.reference = 'ASH'
 
 
-if __name__ == '__main__':
-    unittest.main()
-
-
 def add_mutations(text):
     """ Add a small set of mutations to an RT wild type. """
 
@@ -444,4 +439,8 @@ def add_mutations(text):
     changes = VariantCalls(text)
     for mutation_set in changes:
         seq[mutation_set.pos - 1] = [m.variant for m in mutation_set]
-    return VariantCalls(reference=ref, sample=seq)
+    return VariantCalls(reference=ref, sample=seq)
+
+
+if __name__ == '__main__':
+    unittest.main()