def test_replacement(self):
a = np.arange(10, dtype=float)
a[1] = a[5] = Unknown
ia = preprocess.ReplaceUnknowns(None).transform(a)
np.testing.assert_equal(ia, [0, 0, 2, 3, 4, 0, 6, 7, 8, 9])
a[1] = a[5] = Unknown
ia = preprocess.ReplaceUnknowns(None, value=42).transform(a)
np.testing.assert_equal(ia, [0, 42, 2, 3, 4, 42, 6, 7, 8, 9])
def test_sparse_nans(self):
"""
Remove nans from sparse matrix.
GH-2295
GH-2178
"""
m = sp.csr_matrix(np.ones((3, 3)))
m[0, :] = np.nan
self.assertTrue(np.isnan(m.data).any())
preprocess.ReplaceUnknowns(None, value=42.).transform(m)
self.assertFalse(np.isnan(m.data).any())
def test_sparse(self):
m = sp.csr_matrix(np.eye(10))
rm = preprocess.ReplaceUnknowns(None, value=42).transform(m)
self.assertEqual((m != rm).nnz, 0)
