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)