From 0b466f39d9f34139798b37d9962ee8e756a011be Mon Sep 17 00:00:00 2001
From: Tim O'Donnell <timodonnell@gmail.com>
Date: Wed, 15 Feb 2017 21:57:51 -0500
Subject: [PATCH] fix
---
test/test_antigen_presentation.py | 6 +++---
1 file changed, 3 insertions(+), 3 deletions(-)
diff --git a/test/test_antigen_presentation.py b/test/test_antigen_presentation.py
index 9d2e0674..c154460e 100644
--- a/test/test_antigen_presentation.py
+++ b/test/test_antigen_presentation.py
@@ -3,7 +3,7 @@ import pickle
from nose.tools import eq_, assert_less
import numpy
-from numpy.testing import assert_allclose, assert_array_equal
+from numpy.testing import assert_allclose
import pandas
from mhcflurry import amino_acid
from mhcflurry.antigen_presentation import (
@@ -170,13 +170,13 @@ def test_presentation_model():
peptides.prediction[peptides.hit].mean())
model2 = pickle.loads(pickle.dumps(model))
- assert_array_equal(
+ assert_allclose(
model.predict(peptides), model2.predict(peptides))
model3 = unfit_model.clone()
assert not model3.has_been_fit
model3.restore_fit(model2.get_fit())
- assert_array_equal(
+ assert_allclose(
model.predict(peptides), model3.predict(peptides))
better_unfit_model = models["A_ms + P"]
--
GitLab