def test_coord_input(self): source = AuxCoord(self.src_levels.data) source.metadata = self.src_levels.metadata for axis in self.axes: result = relevel(self.cube, source, [0, 12, 13], axis=axis) self.assertEqual(result.shape, (3, 1, 1)) assert_array_equal(result.data.flatten(), [0, 120, np.nan])
def test_coord_input(self): source = AuxCoord(self.src_levels.data) metadata = self.src_levels.metadata._asdict() metadata["coord_system"] = None metadata["climatological"] = None source.metadata = metadata for axis in self.axes: result = relevel(self.cube, source, [0, 12, 13], axis=axis) self.assertEqual(result.shape, (3, 1, 1)) assert_array_equal(result.data.flatten(), [0, 120, np.nan])