From 6cd2f4f1f2f4a4dca74fcfd6484278cc90e6f77a Mon Sep 17 00:00:00 2001 From: Jonathan Kamens Date: Thu, 12 Mar 2015 10:32:26 -0400 Subject: [PATCH] TEST: Update Security class unit tests for Python3 compatibility --- tests/test_security_object.py | 11 +++++++++-- 1 file changed, 9 insertions(+), 2 deletions(-) diff --git a/tests/test_security_object.py b/tests/test_security_object.py index d6605fb6..528a8bc3 100644 --- a/tests/test_security_object.py +++ b/tests/test_security_object.py @@ -1,3 +1,4 @@ +import sys from unittest import TestCase from zipline.assets._securities import Security @@ -29,5 +30,11 @@ class TestSecurityRichCmp(TestCase): self.assertTrue(Security(5) > Security(4)) def test_type_mismatch(self): - self.assertIsNotNone(Security(3) < 'a') - self.assertIsNotNone('a' < Security(3)) + 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)