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