From 4e68da437496ddf8e6158103e7bd44698bbcd291 Mon Sep 17 00:00:00 2001 From: Tim O'Donnell Date: Mon, 4 Dec 2017 14:19:08 -0500 Subject: [PATCH] Fix predict test --- test/test_predict_command.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/test/test_predict_command.py b/test/test_predict_command.py index bc8969e6..cfaf2a0a 100644 --- 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 <