Merge pull request #1782 from quantopian/no_slippage_comparison

MAINT: Remove __eq__ implementation from slippage
This commit is contained in:
Andrew Liang
2017-05-23 15:39:36 -04:00
committed by GitHub
2 changed files with 1 additions and 20 deletions
-10
View File
@@ -97,16 +97,6 @@ class SlippageTestCase(WithCreateBarData,
super(SlippageTestCase, cls).init_class_fixtures()
cls.ASSET133 = cls.env.asset_finder.retrieve_asset(133)
def test_equality_and_comparison(self):
vol1 = VolumeShareSlippage(volume_limit=0.2)
vol2 = VolumeShareSlippage(volume_limit=0.2)
self.assertEqual(vol1, vol2)
self.assertEqual(hash(vol1), hash(vol2))
self.assertEqual(vol1.__dict__, vol1.asdict())
self.assertEqual(vol2.__dict__, vol2.asdict())
def test_allowed_asset_types(self):
# Custom equities model.
class MyEquitiesModel(EquitySlippageModel):
+1 -10
View File
@@ -19,7 +19,7 @@ import math
import numpy as np
from pandas import isnull
from six import with_metaclass, iteritems
from six import with_metaclass
from toolz import merge
from zipline.assets import Equity, Future
@@ -163,15 +163,6 @@ class SlippageModel(with_metaclass(FinancialModelMeta)):
self._volume_for_bar += abs(txn.amount)
yield order, txn
def __eq__(self, other):
return self.asdict() == other.asdict()
def __hash__(self):
return hash((
type(self),
tuple(sorted(iteritems(self.asdict())))
))
def asdict(self):
return self.__dict__