def test_numpy_array(self): array = NewAxesArray(np.arange(24), [1, 2]) self.assertIsInstance(array.masked_array(), np.ma.MaskedArray) assert_array_equal(array.masked_array(), np.arange(24).reshape(1, 24, 1, 1))