wassname/catalyst
1
0
Fork 0
mirror of https://github.com/wassname/catalyst.git synced 2026-09-11 12:00:50 +08:00
Code Issues Packages Projects Releases Wiki Activity

ENH: Adds Equity and Future classes as extensions of Asset class

Browse Source
This commit is contained in:
jfkirk
2015-03-19 13:23:40 -04:00
parent 2fd2f02c2a
commit 84d4fa3c08
2 changed files with 80 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tests/test_assets.py
+17 -2
View File
@@ -20,7 +20,7 @@ Tests for the zipline.assets package
import sys
from unittest import TestCase
from zipline.assets._assets import Asset
from zipline.assets._assets import Asset, Future
class AssetTestCase(TestCase):
@@ -36,7 +36,7 @@ class AssetTestCase(TestCase):
self.assertEquals(str(Asset(5061)), 'Asset(5061)')
class TestSecurityRichCmp(TestCase):
class TestAssetRichCmp(TestCase):
def test_lt(self):
self.assertTrue(Asset(3) < Asset(4))
@@ -72,3 +72,18 @@ class TestSecurityRichCmp(TestCase):
Asset(3) < 'a'
with self.assertRaises(TypeError):
'a' < Asset(3)
class testFuture(TestCase):
def test_repr(self):
future = Future(2468,
notice_date='2014-01-20',
expiration_date='2014-02-20')
rep = future.__repr__()
self.assertTrue("Future" in rep)
self.assertTrue("2468" in rep)
self.assertTrue("notice_date='2014-01-20'" in rep)
self.assertTrue("expiration_date='2014-02-20'" in rep)