def test_dimension_values_vdim(self):
cube = HoloCube(self.cube, kdims=["longitude", "latitude"])
self.assertEqual(
cube.dimension_values("unknown", flat=False),
np.array([[0, 4, 8], [1, 5, 9], [2, 6, 10], [3, 7, 11]], dtype=np.int32),
)
def test_dimension_values_kdim(self):
cube = HoloCube(self.cube, kdims=["longitude", "latitude"])
self.assertEqual(cube.dimension_values("longitude", expanded=False), np.array([-1, 0, 1, 2], dtype=np.int32))
def test_dimension_values_vdim(self):
cube = HoloCube(self.cube, kdims=['longitude', 'latitude'])
self.assertEqual(cube.dimension_values('unknown', flat=False),
np.array([[ 0, 1, 2, 3],
[ 4, 5, 6, 7],
[ 8, 9, 10, 11]], dtype=np.int32))
def test_dimension_values_kdim_expanded(self):
cube = HoloCube(self.cube, kdims=['longitude', 'latitude'])
self.assertEqual(cube.dimension_values('longitude'),
np.array([-1, -1, -1, 0, 0, 0,
1, 1, 1, 2, 2, 2], dtype=np.int32))
def test_dimension_values_vdim(self):
cube = HoloCube(self.cube, kdims=['longitude', 'latitude'])
self.assertEqual(
cube.dimension_values('unknown', flat=False),
np.array([[0, 4, 8], [1, 5, 9], [2, 6, 10], [3, 7, 11]],
dtype=np.int32))
def test_dimension_values_kdim(self):
cube = HoloCube(self.cube, kdims=['longitude', 'latitude'])
self.assertEqual(cube.dimension_values('longitude', expanded=False),
np.array([-1, 0, 1, 2], dtype=np.int32))
