tests for jq

amyasnikov · Mar 6, 2024 · c1607c5 · c1607c5
1 parent 57dff0b
commit c1607c5
Showing 1 changed file with 12 additions and 1 deletion.
diff --git a/validity/tests/test_utils/test_json.py b/validity/tests/test_utils/test_json.py
@@ -1,4 +1,5 @@
-from validity.utils.json import transform_json
+from validity.utils.json import transform_json, jq
+import pytest
 
 
 class TestTransformJson:
@@ -31,3 +32,13 @@ def test_change_key(self):
         )
         assert result["groups"]["admin2"] == self.JSON["groups"]["admin"]
         assert "admin" not in result["groups"]
+
+
+@pytest.mark.parametrize('data, expression, result', [
+    ({'a': {'b': 'one', 'c': 'two'}}, '. | mkarr(.a.b)', {'a': {'b': ['one'], 'c': 'two'}}),
+    ({'a': {'b': ['one'], 'c': 'two'}}, '. | mkarr(.a.b)', {'a': {'b': ['one'], 'c': 'two'}}),
+    ({'a': '10.2', 'b': {'c': '20'}}, '. | mknum(.b)', {'a': '10.2', 'b': {'c': 20}}),
+    ({'a': '10.2', 'b': {'c': '20'}}, '. | mknum', {'a': 10.2, 'b': {'c': 20}}),
+])
+def test_jq(data, expression, result):
+    assert jq.first(expression, data) == result