mirror of
https://github.com/wassname/catalyst.git
synced 2026-07-08 21:01:50 +08:00
TST: Windows fixups
This commit is contained in:
@@ -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)
|
||||
|
||||
|
||||
@@ -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))
|
||||
|
||||
Reference in New Issue
Block a user