本文整理汇总了Python中zipline.test_algorithms.TestRegisterTransformAlgorithm类的典型用法代码示例。如果您正苦于以下问题：Python TestRegisterTransformAlgorithm类的具体用法？Python TestRegisterTransformAlgorithm怎么用？Python TestRegisterTransformAlgorithm使用的例子？那么恭喜您, 这里精选的类代码示例或许可以为您提供帮助。
在下文中一共展示了TestRegisterTransformAlgorithm类的9个代码示例，这些例子默认根据受欢迎程度排序。您可以为喜欢或者感觉有用的代码点赞，您的评价将有助于系统推荐出更棒的Python代码示例。
示例1: test_df_as_input
def test_df_as_input(self):
algo = TestRegisterTransformAlgorithm(
self.sim_params,
sids=[0, 1]
)
algo.run(self.df)
assert isinstance(algo.sources[0], DataFrameSource)示例2: test_run_twice
def test_run_twice(self):
algo = TestRegisterTransformAlgorithm(sim_params=self.sim_params, sids=[0, 1])
res1 = algo.run(self.df)
res2 = algo.run(self.df)
np.testing.assert_array_equal(res1, res2)示例3: test_multi_source_as_input_no_start_end
def test_multi_source_as_input_no_start_end(self):
algo = TestRegisterTransformAlgorithm(
sids=[133]
)
with self.assertRaises(AssertionError):
algo.run([self.source, self.df_source])示例4: test_transform_registered
def test_transform_registered(self):
algo = TestRegisterTransformAlgorithm(sids=[133])
algo.run(self.source)
assert "mavg" in algo.registered_transforms
assert algo.registered_transforms["mavg"]["args"] == (["price"],)
assert algo.registered_transforms["mavg"]["kwargs"] == {"window_length": 2, "market_aware": True}
assert algo.registered_transforms["mavg"]["class"] is MovingAverage示例5: test_source_as_input
def test_source_as_input(self):
algo = TestRegisterTransformAlgorithm(
self.sim_params,
sids=[133]
)
algo.run(self.source)
self.assertEqual(len(algo.sources), 1)
assert isinstance(algo.sources[0], SpecificEquityTrades)示例6: test_transform_registered
def test_transform_registered(self):
algo = TestRegisterTransformAlgorithm(sids=[133])
algo.run(self.source)
assert 'mavg' in algo.registered_transforms
assert algo.registered_transforms['mavg']['args'] == (['price'],)
assert algo.registered_transforms['mavg']['kwargs'] == \
{'window_length': 2, 'market_aware': True}
assert algo.registered_transforms['mavg']['class'] is MovingAverage示例7: test_multi_source_as_input
def test_multi_source_as_input(self):
sim_params = SimulationParameters(
self.df.index[0],
self.df.index[-1]
)
algo = TestRegisterTransformAlgorithm(
sim_params=sim_params,
sids=[0, 1, 133]
)
algo.run([self.source, self.df_source])
self.assertEqual(len(algo.sources), 2)示例8: test_panel_as_input
def test_panel_as_input(self):
algo = TestRegisterTransformAlgorithm(sim_params=self.sim_params, sids=[0, 1])
algo.run(self.panel)
assert isinstance(algo.sources[0], DataPanelSource)示例9: test_multi_source_as_input
def test_multi_source_as_input(self):
algo = TestRegisterTransformAlgorithm(sids=[0, 1, 133])
algo.run([self.source, self.df_source], start=self.df.index[0], end=self.df.index[-1])
self.assertEqual(len(algo.sources), 2)