From 3dc666cbafbf9346b314a88ec4504d34d7f070bf Mon Sep 17 00:00:00 2001
From: Tim O'Donnell <timodonnell@gmail.com>
Date: Fri, 13 Mar 2020 08:50:43 -0400
Subject: [PATCH] fix test
---
mhcflurry/predict_command.py | 2 +-
test/test_predict_command.py | 3 ++-
2 files changed, 3 insertions(+), 2 deletions(-)
diff --git a/mhcflurry/predict_command.py b/mhcflurry/predict_command.py
index f468b168..cf23a02b 100644
--- a/mhcflurry/predict_command.py
+++ b/mhcflurry/predict_command.py
@@ -278,7 +278,7 @@ def run(argv=sys.argv[1:]):
del predictions["best_allele"]
for col in predictions.columns:
- if col not in ("allele", "peptide", "experiment_name"):
+ if col not in ("allele", "peptide", "experiment_name", "peptide_num"):
df[args.prediction_column_prefix + col] = predictions[col]
if args.out:
diff --git a/test/test_predict_command.py b/test/test_predict_command.py
index a4fbdc51..8d73fa7a 100644
--- a/test/test_predict_command.py
+++ b/test/test_predict_command.py
@@ -45,6 +45,7 @@ def test_no_csv():
"--alleles", "HLA-A0201", "H-2-Kb",
"--peptides", "SIINFEKL", "DENDREKLLL", "PICKLEEE",
"--prediction-column-prefix", "mhcflurry1_",
+ "--affinity-only",
]
deletes = []
@@ -61,7 +62,7 @@ def test_no_csv():
os.unlink(delete)
print(result)
- assert_equal(result.shape, (6, 6))
+ assert_equal(len(result), 6)
sub_result1 = result.loc[result.peptide == "SIINFEKL"].set_index("allele")
print(sub_result1)
assert (
--
GitLab