def test_slice_mask_begin_negative(self):
root = create_expression.create_expression_from_prensor(
prensor_test_util.create_big_prensor())
root_2, new_path = slice_expression._get_slice_mask(
root, path.Path(["doc"]), -1, None)
result = calculate.calculate_prensors([root_2])[0]
self.assertAllEqual(
result.get_descendant_or_error(new_path).node.parent_index,
[0, 1, 1])
self.assertAllEqual(
result.get_descendant_or_error(new_path).node.values,
[True, False, True])
def test_slice_mask_begin_positive(self):
with self.session(use_gpu=False) as sess:
root = create_expression.create_expression_from_prensor(
prensor_test_util.create_big_prensor())
root_2, new_path = slice_expression._get_slice_mask(
root, path.Path(["doc"]), 1, None)
result = prensor_value.materialize(
calculate.calculate_prensors([root_2])[0], sess)
self.assertAllEqual(
result.get_descendant_or_error(new_path).node.parent_index,
[0, 1, 1])
self.assertAllEqual(
result.get_descendant_or_error(new_path).node.values,
[False, False, True])
