Updated tests to not conflict with recent module changes

jaredgk · Aug 3, 2017 · fe4e2bc · fe4e2bc
1 parent d10c77b
commit fe4e2bc
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/andrew/andrew_tests.py b/andrew/andrew_tests.py
@@ -12,13 +12,13 @@ def test_vcf_argument_parser (self):
         self.assertEqual(input_arg, ['--gzvcf', 'example/locus8.vcf.gz'])
 
     def test_produce_vcftools_log (self):
-        vcftools.produce_vcftools_log('Log Test:\n1\n2\n3\n', 'out', '.logTest')
+        vcftools.produce_vcftools_log('Log Test:\n1\n2\n3\n', 'out', 'logTest')
         self.assertTrue(compare_to_expected('out.logTest.log', 'example/locus8.logTest.log'))
         self.addCleanup(os.remove, 'out.logTest.log')
 
     def test_check_vcftools_for_errors (self):
         self.assertTrue(vcftools.check_vcftools_for_errors('Log Test:\n1\n2\n3\nRun Time'))
-        with self.assertRaises(SystemExit) as cm:
+        with self.assertRaises(Exception) as cm:
             vcftools.check_vcftools_for_errors('Log Test:\n1\n2\n3\nError: No Input')
         self.assertEqual(cm.exception.code, 'Error: No Input')