Files
regorus/tests/parser/cases/expressions/array-compr.yaml
Anand Krishnamoorthi 444b2970a1 feat!: Indexes for nodes in the AST (#414)
Indexes allow associating extra data with nodes in the AST
using an array and then quickly looking up the array to fetch
the extra data.

- Index eidx for expressions
- Index sidx for statements
- Index qidx for queries.

AST nodes are not cloneable. Therefore once a module is created,
it is not possible to accidentally create two nodes with the same
index inadvertently via clone.

Also added IndexChecker in debug builds. When a module is parsed,
it will assert that indexes have been constructed correctly.

AST Cleanup
- Make literal expressions (null, val, number, string etc) also structs
  to match all other expressions
- Merge True and False nodes into a single Bool node.

Also update dependencies.

Signed-off-by: Anand Krishnamoorthi <anakrish@microsoft.com>
2025-06-20 15:09:07 -05:00

285 lines
7.4 KiB
YAML

# Copyright (c) Microsoft Corporation.
# Licensed under the MIT License.
cases:
- note: case1-compr
rego: |
package test
x = {1}
y = {2}
z = [ x | y ] #, 5] # in {5}, 6]
num_expressions: 11
num_queries: 1
num_statements: 1
policy:
- --skip--
- --skip--
- spec:
head:
compr:
refr:
var: z
eidx: 7
assign:
op: =
value:
arraycompr:
term:
var: x
eidx: 8
query:
qidx: 0
stmts:
- span: y
literal:
expr:
var: y
eidx: 9
sidx: 0
eidx: 10
bodies: []
- note: case2-array
rego: |
package test
x = {1}
y = {2}
z = [ x | y, 5] # in {5}, 6]
num_expressions: 13
num_queries: 0
num_statements: 0
policy:
- --skip--
- --skip--
- spec:
head:
compr:
refr:
var: z
eidx: 7
assign:
op: =
value:
array:
- binexpr:
op: "|"
lhs:
var: x
eidx: 8
rhs:
var: y
eidx: 9
eidx: 10
- number: 5
eidx: 11
eidx: 12
bodies: []
- note: case3-compr
rego: |
package test
import future.keywords.in
x = {1}
y = {2}
z = [ x | y, 5 in {5}] #, 6]
num_expressions: 18
num_queries: 1
num_statements: 1
policy:
- --skip--
- --skip--
- spec:
head:
compr:
refr:
var: z
eidx: 10
assign:
op: =
value:
arraycompr:
term:
var: x
eidx: 11
query:
qidx: 0
stmts:
- span: y, 5 in {5}
sidx: 0
literal:
expr:
inexpr:
key:
var: y
eidx: 12
value:
number: 5
eidx: 13
collection:
set:
- number: 5
eidx: 14
eidx: 15
eidx: 16
eidx: 15
eidx: 17
bodies: []
- note: case4-array
rego: |
package test
import future.keywords.in
x = {1}
y = {2}
z = [ x | y, 5 in {5}, 6]
num_expressions: 20
num_queries: 0
num_statements: 0
policy:
- --skip--
- --skip--
- spec:
head:
compr:
refr:
var: z
eidx: 10
assign:
op: =
value:
array:
- binexpr:
op: "|"
lhs:
var: x
eidx: 11
rhs:
var: y
eidx: 12
eidx: 13
- inexpr:
value:
number: 5
eidx: 14
collection:
set:
- number: 5
eidx: 15
eidx: 16
eidx: 17
- number: 6
eidx: 18
eidx: 19
bodies: []
- note: case5-array
rego: |
package test
import future.keywords.in
x = {1}
y = {2}
z = [ x - {10} | y, 5 in {5}] #, 6]
num_expressions: 22
num_queries: 0
num_statements: 0
policy:
- --skip--
- --skip--
- spec:
head:
compr:
refr:
var: z
eidx: 10
assign:
op: =
value:
array:
- binexpr:
op: "|"
lhs:
arithexpr:
op: "-"
lhs:
var: x
eidx: 11
rhs:
set:
- number: 10
eidx: 12
eidx: 13
eidx: 14
rhs:
var: y
eidx: 15
eidx: 16
- inexpr:
value:
number: 5
eidx: 17
collection:
set:
- number: 5
eidx: 18
eidx: 19
eidx: 20
eidx: 21
bodies: []
- note: case6-compr
rego: |
package test
import future.keywords.in
x = {1}
y = {2}
z = [ (x - {10}) | y, 5 in {5}] #, 6]
num_expressions: 21
num_queries: 1
num_statements: 1
policy:
- --skip--
- --skip--
- spec:
head:
compr:
refr:
var: z
eidx: 10
assign:
op: =
value:
arraycompr:
term:
arithexpr:
op: "-"
lhs:
var: x
eidx: 11
rhs:
set:
- number: 10
eidx: 12
eidx: 13
eidx: 14
query:
qidx: 0
stmts:
- sidx: 0
literal:
expr:
inexpr:
key:
var: y
eidx: 15
value:
number: 5
eidx: 16
collection:
set:
- number: 5
eidx: 17
eidx: 18
eidx: 19
eidx: 20
bodies: []