fixed capitalization issue for postgres

kgmcquate · Jan 3, 2024 · 1472995 · 1472995
1 parent 2056d20
commit 1472995
Show file tree

Hide file tree

Showing 3 changed files with 7 additions and 7 deletions.
diff --git a/macros/test_generation/get_accepted_values_test_suggestions.sql b/macros/test_generation/get_accepted_values_test_suggestions.sql
@@ -101,10 +101,10 @@
         {# {{ print(cardinality_result.COLNAME) }} #}
 
         {% set col_config = {
-                "name": cardinality_result.COLNAME,
+                "name": cardinality_result[1],
                 "description": "Accepted values test generated by dbt-testgen",
                 "tests": [
-                    {"accepted_values": {"values": fromjson(cardinality_result.UNIQUE_VALUES)|sort}}
+                    {"accepted_values": {"values": fromjson(cardinality_result[3])|sort}}
                 ]
             }
         %}

diff --git a/macros/test_generation/get_range_test_suggestions.sql b/macros/test_generation/get_range_test_suggestions.sql
@@ -67,13 +67,13 @@
     {% set column_tests = [] %}
     {% for result in results %}
         {% set col_config = {
-                "name": result.COLNAME,
+                "name": result[0],
                 "description": "Numeric range test generated by dbt-testgen",
                 "tests": [
                     {
                         "dbt_utils.accepted_range": {
-                            "min_value": result.COL_MIN,
-                            "max_value": result.COL_MAX
+                            "min_value": result[1],
+                            "max_value": result[2]
                         }
                     }
                 ]

diff --git a/macros/test_generation/get_uniqueness_test_suggestions.sql b/macros/test_generation/get_uniqueness_test_suggestions.sql
@@ -100,7 +100,7 @@
         {{ "SELECT count(1) AS TABLE_COUNT FROM " ~ table_relation }} 
     {% endset%}
 
-    {% set table_count = testgen.query_as_list(count_sql)[0].TABLE_COUNT %}
+    {% set table_count = testgen.query_as_list(count_sql)[0][0] %}
 
     {% set cardinality_results = zip(column_combinations, testgen.query_as_list(count_distinct_sql)) %}
 
@@ -109,7 +109,7 @@
 
     {% set unique_keys = [] %}
     {% for cardinality_result in cardinality_results %}
-        {% if cardinality_result[1].CARDINALITY == table_count %}
+        {% if cardinality_result[1][1] == table_count %}
             {# {{ print(cardinality_result) }} #}
             {% do unique_keys.append(cardinality_result[0]) %}
         {% endif %}