BUG: Enable comparison of an Asset to an int64 on 32-bit python

We use a number of mappings keyed by int64, which otherwise raised
KeyErrors for Assets.
This commit is contained in:
Richard Frank
2016-01-21 11:47:29 -05:00
parent daf05c6b59
commit 2eeb6e78f6
2 changed files with 18 additions and 2 deletions
+10 -2
View File
@@ -29,7 +29,7 @@ from pandas.tseries.tools import normalize_date
from pandas.util.testing import assert_frame_equal
from nose_parameterized import parameterized
from numpy import full
from numpy import full, int32, int64
import sqlalchemy as sa
from zipline.assets import (
@@ -39,7 +39,7 @@ from zipline.assets import (
AssetFinder,
AssetFinderCachedEquities,
)
from six import itervalues
from six import itervalues, integer_types
from toolz import valmap
from zipline.assets.futures import (
@@ -212,6 +212,14 @@ class AssetTestCase(TestCase):
self.assertEqual(s_23, s_23)
self.assertEqual(s_23, 23)
self.assertEqual(23, s_23)
self.assertEqual(int32(23), s_23)
self.assertEqual(int64(23), s_23)
self.assertEqual(s_23, int32(23))
self.assertEqual(s_23, int64(23))
# Check all int types (includes long on py2):
for int_type in integer_types:
self.assertEqual(int_type(23), s_23)
self.assertEqual(s_23, int_type(23))
self.assertNotEqual(s_23, s_24)
self.assertNotEqual(s_23, 24)