Fix predict test

openvax · Dec 4, 2017 · 4e68da4 · 4e68da4
1 parent 0f9394e
commit 4e68da4
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/test/test_predict_command.py b/test/test_predict_command.py
@@ -32,7 +32,7 @@ def test_csv():
         for delete in deletes:
             os.unlink(delete)
 
-    assert_equal(result.shape, (3, 6))
+    assert_equal(result.shape, (3, 7))
 
 
 def test_no_csv():
@@ -55,7 +55,8 @@ def test_no_csv():
         for delete in deletes:
             os.unlink(delete)
 
-    assert_equal(result.shape, (6, 5))
+    print(result)
+    assert_equal(result.shape, (6, 6))
     sub_result1 = result.ix[result.peptide == "SIINFEKL"].set_index("allele")
     assert (
         sub_result1.ix["H-2-Kb"].mhcflurry1_prediction <