updated based on PR feedback.

This commit is contained in:
fawce
2012-09-30 22:12:00 -04:00
parent ebde40ab05
commit 7a8697f0e5
8 changed files with 80 additions and 47 deletions
-4
View File
@@ -303,10 +303,6 @@ class BatchTransformTestCase(TestCase):
# test overloaded class
for test_history in [algo.history_return_price_class, algo.history_return_price_decorator]:
self.assertTrue(np.all(test_history[2].values.flatten() == range(4, 10)))
self.assertTrue(np.all(test_history[3].values.flatten() == range(4, 10)))
self.assertTrue(np.all(test_history[4].values.flatten() == range(6, 14)))
np.testing.assert_array_equal(
range(4, 10),
test_history[2].values.flatten()