mirror of
https://github.com/wassname/catalyst.git
synced 2026-06-30 13:27:48 +08:00
Jonathan Kamens
716bdbb7cd
The >= comparison for the Cythonized Security object was actually doing <=. Fix this and add unit tests for all the Security object rich comparison operators.
30 lines
1018 B
Python
30 lines
1018 B
Python
from unittest import TestCase
|
|
from zipline.assets._securities import Security
|
|
|
|
|
|
class TestSecurityRichCmp(TestCase):
|
|
def test_lt(self):
|
|
self.assertTrue(Security(3) < Security(4))
|
|
self.assertFalse(Security(4) < Security(4))
|
|
self.assertFalse(Security(5) < Security(4))
|
|
|
|
def test_le(self):
|
|
self.assertTrue(Security(3) <= Security(4))
|
|
self.assertTrue(Security(4) <= Security(4))
|
|
self.assertFalse(Security(5) <= Security(4))
|
|
|
|
def test_eq(self):
|
|
self.assertFalse(Security(3) == Security(4))
|
|
self.assertTrue(Security(4) == Security(4))
|
|
self.assertFalse(Security(5) == Security(4))
|
|
|
|
def test_ge(self):
|
|
self.assertFalse(Security(3) >= Security(4))
|
|
self.assertTrue(Security(4) >= Security(4))
|
|
self.assertTrue(Security(5) >= Security(4))
|
|
|
|
def test_gt(self):
|
|
self.assertFalse(Security(3) > Security(4))
|
|
self.assertFalse(Security(4) > Security(4))
|
|
self.assertTrue(Security(5) > Security(4))
|