From c31d3b7904fc7b65faf087e825c33694f5c4b320 Mon Sep 17 00:00:00 2001 From: Gil Wassermann Date: Thu, 21 Jul 2016 12:25:02 -0400 Subject: [PATCH] ENH: Test now works --- tests/pipeline/test_filter.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/tests/pipeline/test_filter.py b/tests/pipeline/test_filter.py index 08eab31c..a527cee5 100644 --- a/tests/pipeline/test_filter.py +++ b/tests/pipeline/test_filter.py @@ -393,7 +393,7 @@ class FilterTestCase(BasePipelineTestCase): inputs = () filter_results = self.run_graph( - TermGraph({'filter': nansum(CustomFilterBool, axis=0)}), - initial_workspace={CustomFilterBool: data}, + TermGraph({'filter': CustomFilterBool()}), + initial_workspace={CustomFilterBool(): data}, ) - check_arrays(filter_results['filter'], array([1, 1, 0, 2])) + check_arrays(nansum(filter_results['filter'], axis=0), array([1, 1, 0, 2]))