---
cases:
  - data: {}
    modules:
      - |
        package generated

        p = x {
          s1 = {1, 2, 3}
          s2 = {2}
          union({s1, s2}, x)
        }
    note: union/union_2_sets
    query: data.generated.p = x
    want_result:
      - x:
          - 1
          - 2
          - 3
