def test5(self):
dimension_values = [
[0, 1, 2, 3, 4, 5, 6, 7, 8,9 ]
]
split_dimension_values = indexing.split_numpy_index_over_dimensions(slice(-2,10), dimension_values)
self.assertEquals(split_dimension_values[0], [8, 9])
split_dimension_values = indexing.split_numpy_index_over_dimensions(slice(-3,3,-1), dimension_values)
self.assertEquals(split_dimension_values[0], [7, 6, 5, 4])
def test4(self):
dimension_values = [
[0, 1, 2, 3, 4, 5, 6, 7, 8, ]
]
split_dimension_values = indexing.split_numpy_index_over_dimensions(slice(1,7,2), dimension_values)
self.assertEquals(len(split_dimension_values) ,1)
self.assertEquals(split_dimension_values[0], [1, 3, 5])
def test7(self):
dimension_values = [[0, 1], [0, 1, 2], [0]]
split_dimension_values = indexing.split_numpy_index_over_dimensions(
slice(1, 2), dimension_values)
self.assertEqual(split_dimension_values[0], [1])
self.assertEqual(split_dimension_values[1], [0, 1, 2])
self.assertEqual(split_dimension_values[2], [0])
def test3(self):
dimension_values = [[3, 4, 5, 6], ['a', 'b', 'c']]
split_dimension_values = indexing.split_numpy_index_over_dimensions(
slice(0, 2), dimension_values)
self.assertEqual(len(split_dimension_values), 2)
self.assertEqual(split_dimension_values[0], [3, 4])
self.assertEqual(split_dimension_values[1], ['a', 'b', 'c'])
def test3(self):
dimension_values = [
[3,4,5,6],
['a','b','c']
]
split_dimension_values = indexing.split_numpy_index_over_dimensions(slice(0,2), dimension_values)
self.assertEquals(len(split_dimension_values) ,2)
self.assertEquals(split_dimension_values[0], [3,4])
self.assertEquals(split_dimension_values[1], ['a','b','c'])
def test2(self):
dimension_values = [
[3,4,5,6],
['a','b','c']
]
split_dimension_values = indexing.split_numpy_index_over_dimensions((1,2), dimension_values)
self.assertEquals(len(split_dimension_values) ,2)
self.assertEquals(split_dimension_values[0], 4)
self.assertEquals(split_dimension_values[1], 'c')
def test7(self):
dimension_values = [
[0, 1],
[0, 1, 2],
[0]
]
split_dimension_values = indexing.split_numpy_index_over_dimensions(slice(1,2), dimension_values)
self.assertEquals(split_dimension_values[0], [1])
self.assertEquals(split_dimension_values[1], [0,1,2])
self.assertEquals(split_dimension_values[2], [0])
def test8(self):
dimension_values = [[0, 1], [0, 1, 2], [0]]
split_dimension_values = indexing.split_numpy_index_over_dimensions(
(Ellipsis, 0), dimension_values)
self.assertEqual(split_dimension_values[0], [0, 1])
self.assertEqual(split_dimension_values[1], [0, 1, 2])
self.assertEqual(split_dimension_values[2], 0)
split_dimension_values = indexing.split_numpy_index_over_dimensions(
(slice(None), slice(None), 0), dimension_values)
self.assertEqual(split_dimension_values[0], [0, 1])
self.assertEqual(split_dimension_values[1], [0, 1, 2])
self.assertEqual(split_dimension_values[2], 0)
split_dimension_values = indexing.split_numpy_index_over_dimensions(
(Ellipsis, 0, Ellipsis), dimension_values)
self.assertEqual(split_dimension_values[0], [0, 1])
self.assertEqual(split_dimension_values[1], 0)
self.assertEqual(split_dimension_values[2], [0])
def test8(self):
dimension_values = [
[0, 1],
[0, 1, 2],
[0]
]
split_dimension_values = indexing.split_numpy_index_over_dimensions((Ellipsis,0), dimension_values)
self.assertEquals(split_dimension_values[0], [0,1])
self.assertEquals(split_dimension_values[1], [0,1,2])
self.assertEquals(split_dimension_values[2], 0)
split_dimension_values = indexing.split_numpy_index_over_dimensions((slice(None),slice(None),0), dimension_values)
self.assertEquals(split_dimension_values[0], [0,1])
self.assertEquals(split_dimension_values[1], [0,1,2])
self.assertEquals(split_dimension_values[2], 0)
split_dimension_values = indexing.split_numpy_index_over_dimensions((Ellipsis,0, Ellipsis), dimension_values)
self.assertEquals(split_dimension_values[0], [0,1])
self.assertEquals(split_dimension_values[1], 0)
self.assertEquals(split_dimension_values[2], [0])
def test6(self):
dimension_values = [
[0, 1, 2, 3, 4, 5, 6, 7, 8,9 ]
]
split_dimension_values = indexing.split_numpy_index_over_dimensions(slice(5,None), dimension_values)
self.assertEquals(split_dimension_values[0], [5, 6, 7, 8, 9])
