示例#1
0
 def test_index_of_single_item_c(self):
     a = W_NDimArray(10 * 5 * 3, [10, 5, 3], MockDtype(), 'C')
     r = a._index_of_single_item(self.space, self.newtuple(1, 2, 2))
     assert r == 1 * 3 * 5 + 2 * 3 + 2
     s = a.create_slice([(0, 10, 1, 10), (2, 0, 0, 1)])
     r = s._index_of_single_item(self.space, self.newtuple(1, 0))
     assert r == a._index_of_single_item(self.space, self.newtuple(1, 2, 0))
     r = s._index_of_single_item(self.space, self.newtuple(1, 1))
     assert r == a._index_of_single_item(self.space, self.newtuple(1, 2, 1))
示例#2
0
 def test_index_of_single_item_c(self):
     a = W_NDimArray(10 * 5 * 3, [10, 5, 3], MockDtype(), 'C')
     r = a._index_of_single_item(self.space, self.newtuple(1, 2, 2))
     assert r == 1 * 3 * 5 + 2 * 3 + 2
     s = a.create_slice([(0, 10, 1, 10), (2, 0, 0, 1)])
     r = s._index_of_single_item(self.space, self.newtuple(1, 0))
     assert r == a._index_of_single_item(self.space, self.newtuple(1, 2, 0))
     r = s._index_of_single_item(self.space, self.newtuple(1, 1))
     assert r == a._index_of_single_item(self.space, self.newtuple(1, 2, 1))