wassname/catalyst
1
0
Fork 0
mirror of https://github.com/wassname/catalyst.git synced 2026-06-30 19:23:51 +08:00
Code Issues Packages Projects Releases Wiki Activity

TEST: test for rolling nature of smoothing filter

Browse Source
This commit is contained in:
Gil Wassermann
2016-08-01 15:35:22 -04:00
parent 7623c0f6eb
commit 574d7b197f
1 changed files with 12 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tests/pipeline/test_filter.py
+12 -5
View File
@@ -410,7 +410,7 @@ class FilterTestCase(BasePipelineTestCase):
strictly_true_filter = StrictlyTrueFilter(
inputs=(InputFilter(), ),
window_length=self.default_shape[0]
window_length=(self.default_shape[0]-1)
)
results = self.run_graph(
@@ -418,11 +418,18 @@ class FilterTestCase(BasePipelineTestCase):
initial_workspace={InputFilter(): data}
)
expected_result = full(self.default_shape[1], False, dtype=bool)
expected_result[0] = True
expected_result_0 = full(self.default_shape[1], False, dtype=bool)
expected_result_0[0] = True
check_arrays(
reshape(results['Filter'], expected_result.shape[0]),
expected_result,
reshape(results['Filter'][0], expected_result_0.shape[0]),
expected_result_0,
)
expected_result_1 = full(self.default_shape[1], False, dtype=bool)
expected_result_1[:2] = True
check_arrays(
reshape(results['Filter'][1], expected_result_1.shape[0]),
expected_result_1,
)
@parameter_space(factor_len=[2, 3, 4])