Esempio n. 1
0
 def is_restricted(self, assets, dt):
     securities_in_list = self.current_securities(dt)
     if isinstance(assets, Asset):
         return assets in securities_in_list
     return pd.Series(index=pd.Index(assets),
                      data=vectorized_is_element(assets,
                                                 securities_in_list))
Esempio n. 2
0
 def is_restricted(self, assets, dt):
     securities_in_list = self.current_securities(dt)
     if isinstance(assets, Asset):
         return assets in securities_in_list
     return pd.Series(
         index=pd.Index(assets),
         data=vectorized_is_element(assets, securities_in_list)
     )
Esempio n. 3
0
 def is_restricted(self, assets, dt):
     """
     An asset is restricted for all dts if it is in the static list.
     """
     if isinstance(assets, Asset):
         return assets in self._restricted_set
     return pd.Series(index=pd.Index(assets),
                      data=vectorized_is_element(assets,
                                                 self._restricted_set))
Esempio n. 4
0
 def is_restricted(self, assets, dt):
     """
     An asset is restricted for all dts if it is in the static list.
     """
     if isinstance(assets, Asset):
         return assets in self._restricted_set
     return pd.Series(
         index=pd.Index(assets),
         data=vectorized_is_element(assets, self._restricted_set)
     )