|
12 | 12 | from Orange.data import \ |
13 | 13 | Instance, Domain, Unknown, Value, \ |
14 | 14 | DiscreteVariable, ContinuousVariable, StringVariable |
| 15 | +from Orange.tests import assert_array_nanequal |
15 | 16 |
|
16 | 17 |
|
17 | 18 | class TestInstance(unittest.TestCase): |
@@ -75,11 +76,10 @@ def test_init_xym_no_data(self): |
75 | 76 | self.assertEqual(inst._metas.shape, (3, )) |
76 | 77 | self.assertTrue(all(isnan(x) for x in inst._x)) |
77 | 78 | self.assertTrue(all(isnan(x) for x in inst._y)) |
78 | | - with warnings.catch_warnings(): |
79 | | - warnings.simplefilter("ignore", FutureWarning) |
80 | | - assert_array_equal(inst._metas, |
81 | | - np.array([var.Unknown for var in domain.metas], |
82 | | - dtype=object)) |
| 79 | + |
| 80 | + assert_array_nanequal(inst._metas, |
| 81 | + np.array([var.Unknown for var in domain.metas], |
| 82 | + dtype=object)) |
83 | 83 |
|
84 | 84 | def test_init_x_arr(self): |
85 | 85 | domain = self.create_domain(["x", DiscreteVariable("g", values="MF")]) |
@@ -162,12 +162,11 @@ def test_init_inst(self): |
162 | 162 | domain.class_vars, |
163 | 163 | [self.metas[0], "w", domain[0]]) |
164 | 164 | inst2 = Instance(domain2, inst) |
165 | | - with warnings.catch_warnings(): |
166 | | - warnings.simplefilter("ignore", FutureWarning) |
167 | | - assert_array_equal(inst2._x, np.array([Unknown, 0, 43])) |
168 | | - self.assertEqual(inst2._y[0], 1) |
169 | | - assert_array_equal(inst2._metas, np.array([0, Unknown, 42], |
170 | | - dtype=object)) |
| 165 | + |
| 166 | + assert_array_nanequal(inst2._x, np.array([Unknown, 0, 43])) |
| 167 | + self.assertEqual(inst2._y[0], 1) |
| 168 | + assert_array_nanequal(inst2._metas, np.array([0, Unknown, 42], |
| 169 | + dtype=object)) |
171 | 170 |
|
172 | 171 | def test_get_item(self): |
173 | 172 | domain = self.create_domain(["x", DiscreteVariable("g", values="MF")], |
|
0 commit comments