From 59e3b5595af7aabf792ba82e2460bbe615e8dcfb Mon Sep 17 00:00:00 2001
From: Alex Rubinsteyn <alex.rubinsteyn@gmail.com>
Date: Fri, 6 May 2016 16:37:41 -0400
Subject: [PATCH] added failing test for loading Dataset and then training
predictor
---
test/test_class1_binding_predictor_A0205.py | 23 +++++++++++++++++++++
1 file changed, 23 insertions(+)
create mode 100644 test/test_class1_binding_predictor_A0205.py
diff --git a/test/test_class1_binding_predictor_A0205.py b/test/test_class1_binding_predictor_A0205.py
new file mode 100644
index 00000000..0383c40c
--- /dev/null
+++ b/test/test_class1_binding_predictor_A0205.py
@@ -0,0 +1,23 @@
+from mhcflurry.dataset import Dataset
+from mhcflurry.paths import CLASS1_DATA_CSV_PATH
+from mhcflurry import Class1BindingPredictor
+
+from nose.tools import eq_
+import numpy as np
+
+
+def class1_binding_predictor_A0205_training_accuracy():
+
+ dataset = Dataset.from_csv(CLASS1_DATA_CSV_PATH)
+ dataset_a0205 = dataset.get_allele("HLA-A0205")
+
+ predictor = Class1BindingPredictor.from_hyperparameters(name="A0205")
+ predictor.fit_dataset(dataset_a0205)
+ peptides = dataset_a0205.peptides
+ ic50_pred = predictor.predict(peptides)
+ ic50_true = dataset_a0205.affinities
+ eq_(len(ic50_pred), len(ic50_true))
+ assert np.allclose(ic50_pred, ic50_true)
+
+if __name__ == "__main__":
+ class1_binding_predictor_A0205_training_accuracy()
--
GitLab