wassname/catalyst
1
0
Fork 0
mirror of https://github.com/wassname/catalyst.git synced 2026-06-27 23:37:55 +08:00
Code Issues Packages Projects Releases Wiki Activity

TEST: Test forwarding of missing_value.

Browse Source
This commit is contained in:
Scott Sanderson
2016-01-20 22:57:02 -05:00
parent c105735574
commit 09be7acaa8
1 changed files with 14 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tests/pipeline/test_term.py
+14 -4
View File
@@ -18,6 +18,7 @@ from zipline.pipeline.data.testing import TestingDataSet
from zipline.pipeline.term import AssetExists, NotSpecified
from zipline.pipeline.expression import NUMEXPR_MATH_FUNCS
from zipline.utils.numpy_utils import (
bool_dtype,
datetime64ns_dtype,
float64_dtype,
int64_dtype,
@@ -335,10 +336,19 @@ class ObjectIdentityTestCase(TestCase):
SomeFactor(dtype=1)
def test_latest_on_different_dtypes(self):
self.assertIsInstance(TestingDataSet.bool_col.latest, Filter)
self.assertIsInstance(TestingDataSet.float_col.latest, Factor)
self.assertIsInstance(TestingDataSet.datetime_col.latest, Factor)
self.assertIsInstance(TestingDataSet.int_col.latest, Factor)
factor_dtypes = (int64_dtype, float64_dtype, datetime64ns_dtype)
for column in TestingDataSet.columns:
if column.dtype == bool_dtype:
self.assertIsInstance(column.latest, Filter)
elif column.dtype in factor_dtypes:
self.assertIsInstance(column.latest, Factor)
else:
self.fail(
"Unknown dtype %s for column %s" % (column.dtype, column)
)
# These should be the same value, plus this has the convenient
# property of correctly handling `NaN`.
self.assertIs(column.missing_value, column.latest.missing_value)
def test_failure_timing_on_bad_missing_values(self):