mirror of
https://github.com/wassname/catalyst.git
synced 2026-06-28 20:05:52 +08:00
Jonathan Kamens
41 lines
1.4 KiB
Python
41 lines
1.4 KiB
Python
import sys
|
|
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))
|
|
|
|
def test_type_mismatch(self):
|
|
if sys.version_info.major < 3:
|
|
self.assertIsNotNone(Security(3) < 'a')
|
|
self.assertIsNotNone('a' < Security(3))
|
|
else:
|
|
with self.assertRaises(TypeError):
|
|
Security(3) < 'a'
|
|
with self.assertRaises(TypeError):
|
|
'a' < Security(3)
|