From 4e68da437496ddf8e6158103e7bd44698bbcd291 Mon Sep 17 00:00:00 2001
From: Tim O'Donnell <timodonnell@gmail.com>
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 <
--
GitLab