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))