mirror of
https://github.com/wassname/catalyst.git
synced 2026-06-29 01:57:31 +08:00
TEST: Ensure that test_transforms's algos are passed sim_params.
This commit is contained in:
@@ -293,7 +293,7 @@ class TestBatchTransform(TestCase):
|
||||
factory.create_test_df_source(self.sim_params)
|
||||
|
||||
def test_event_window(self):
|
||||
algo = BatchTransformAlgorithm()
|
||||
algo = BatchTransformAlgorithm(sim_params=self.sim_params)
|
||||
algo.run(self.source)
|
||||
wl = algo.window_length
|
||||
# The following assertion depend on window length of 3
|
||||
@@ -358,7 +358,8 @@ class TestBatchTransform(TestCase):
|
||||
)
|
||||
|
||||
def test_passing_of_args(self):
|
||||
algo = BatchTransformAlgorithm(1, kwarg='str')
|
||||
algo = BatchTransformAlgorithm(1,
|
||||
kwarg='str', sim_params=self.sim_params)
|
||||
self.assertEqual(algo.args, (1,))
|
||||
self.assertEqual(algo.kwargs, {'kwarg': 'str'})
|
||||
|
||||
|
||||
Reference in New Issue
Block a user