wassname/catalyst
1
0
Fork 0
mirror of https://github.com/wassname/catalyst.git synced 2026-08-15 12:15:22 +08:00
Code Issues Packages Projects Releases Wiki Activity

ENH: Add isnan, notnan, and isfinite Factor methods.

Browse Source
This commit is contained in:
Scott Sanderson
2015-11-18 21:44:53 -05:00
parent 0cdd88a069
commit b43c4f4c0b
4 changed files with 89 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tests/pipeline/test_filter.py
+36
View File
@@ -11,6 +11,8 @@ from numpy import (
eye,
float64,
full_like,
inf,
isfinite,
nan,
nanpercentile,
ones,
@@ -333,3 +335,37 @@ class FilterTestCase(BasePipelineTestCase):
dtype=bool,
)
check_arrays(results['with'], expected_with)
def test_isnan(self):
data = self.randn_data(seed=10)
diag = eye(*data.shape, dtype=bool)
data[diag] = nan
results = self.run_graph(
TermGraph({'isnan': self.f.isnan()}),
initial_workspace={self.f: data},
)
check_arrays(results['isnan'], diag)
def test_notnan(self):
data = self.randn_data(seed=10)
diag = eye(*data.shape, dtype=bool)
data[diag] = nan
results = self.run_graph(
TermGraph({'notnan': self.f.notnan()}),
initial_workspace={self.f: data},
)
check_arrays(results['notnan'], ~diag)
def test_isfinite(self):
data = self.randn_data(seed=10)
data[:, 0] = nan
data[:, 2] = inf
data[:, 4] = -inf
results = self.run_graph(
TermGraph({'isfinite': self.f.isfinite()}),
initial_workspace={self.f: data},
)
check_arrays(results['isfinite'], isfinite(data))