Function test_json_get_array_index

python/pathway/tests/test_json.py:212–223

()

def test_json_get_array_index():
    input = _json_table(
        index=[0, 1, 2],
        data=[{"field": [1, 2, 3]}, {"field": [4, 5, 6]}, {"field": [7, 8, 9]}],
    )

    result = input.select(result=pw.this.data["field"][pw.this.index.as_int()])

    assert_table_equality(
        _json_table(result=[1, 5, 9]),
        result,
    )


@pytest.mark.parametrize("index", [-1, -4, 3])