From 335f73e1e9b9e4d15b5c0e2c661aba6883b73e8b Mon Sep 17 00:00:00 2001 From: Richard Frank Date: Mon, 7 Mar 2016 09:43:26 -0500 Subject: [PATCH] TST: Windows fixups --- tests/pipeline/test_adjusted_array.py | 2 +- tests/pipeline/test_numerical_expression.py | 4 ++-- 2 files changed, 3 insertions(+), 3 deletions(-) diff --git a/tests/pipeline/test_adjusted_array.py b/tests/pipeline/test_adjusted_array.py index a5fab747..a2a8cde0 100644 --- a/tests/pipeline/test_adjusted_array.py +++ b/tests/pipeline/test_adjusted_array.py @@ -365,7 +365,7 @@ class AdjustedArrayTestCase(TestCase): frame[0, 0] = 5.0 def test_bad_input(self): - msg = "Mask shape \(2, 3\) != data shape \(5, 5\)" + msg = "Mask shape \(2L?, 3L?\) != data shape \(5L?, 5L?\)" data = arange(25).reshape(5, 5) bad_mask = array([[0, 1, 1], [0, 0, 1]], dtype=bool) diff --git a/tests/pipeline/test_numerical_expression.py b/tests/pipeline/test_numerical_expression.py index 040ed124..9922004d 100644 --- a/tests/pipeline/test_numerical_expression.py +++ b/tests/pipeline/test_numerical_expression.py @@ -34,7 +34,7 @@ from zipline.pipeline.expression import ( NumericalExpression, NUMEXPR_MATH_FUNCS, ) -from zipline.testing import check_arrays +from zipline.testing import check_allclose from zipline.utils.numpy_utils import datetime64ns_dtype, float64_dtype @@ -91,7 +91,7 @@ class NumericalExpressionTestCase(TestCase): self.mask.columns, self.mask.values, ) - check_arrays(result, expected) + check_allclose(result, expected) def check_constant_output(self, expr, expected): self.assertFalse(isnan(expected))