From 21b33f03f99a39e33d06cadbb32d20730887503f Mon Sep 17 00:00:00 2001 From: Gil Wassermann Date: Thu, 21 Jul 2016 10:21:19 -0400 Subject: [PATCH] ENH: test filter up and running --- tests/pipeline/test_filter.py | 11 +++++------ 1 file changed, 5 insertions(+), 6 deletions(-) diff --git a/tests/pipeline/test_filter.py b/tests/pipeline/test_filter.py index 9c1cbdae..08eab31c 100644 --- a/tests/pipeline/test_filter.py +++ b/tests/pipeline/test_filter.py @@ -18,7 +18,7 @@ from numpy import ( ones, ones_like, putmask, - sum + nansum ) from numpy.random import randn, seed as random_seed @@ -380,11 +380,10 @@ class FilterTestCase(BasePipelineTestCase): initial_workspace={self.f: data}, ) check_arrays(results['isfinite'], isfinite(data)) - + def test_window_safe(self): """ - Rationale : CustomFactors are *not all* window safe, while - CustomFilters are + This computation is now possible as filters are windowsafe """ data = array([[True, False, False, True], [False, True, False, True]], dtype=bool) @@ -394,7 +393,7 @@ class FilterTestCase(BasePipelineTestCase): inputs = () filter_results = self.run_graph( - TermGraph({'factor': sum(CustomFilterBool, axis=0)}), + TermGraph({'filter': nansum(CustomFilterBool, axis=0)}), initial_workspace={CustomFilterBool: data}, ) - check_arrays(filter_results['factor'], array([1, 1, 0, 2])) + check_arrays(filter_results['filter'], array([1, 1, 0, 2]))