Updates to Dependencies and tests

Author: ParamThakkar123

Project.toml

@@ -4,8 +4,10 @@ authors = ["ParamThakkar123 <[email protected]> and TheCedarPrince <jaco
 version = "0.0.1"
 
 [deps]
+DataFrames = "a93c6f00-e57d-5684-b7b6-d8193f3e46c0"
 DrWatson = "634d3b9d-ee7a-5ddf-bec9-22491ea816e1"
 DuckDB = "d2f5444f-75bc-4fdf-ac35-56f514c445e1"
+FunSQL = "cf6cc811-59f4-4a10-b258-a8547a8f6407"
 HuggingFaceHub = "d0076355-e2c0-48e6-a044-05906e51b7fc"
 JSON3 = "0f8b85d8-7281-11e9-16c2-39a750bddbf1"
 LibPQ = "194296ae-ab2e-5f79-8cd4-7183a0a5a0d1"
@@ -17,8 +19,10 @@ SparseArrays = "2f01184e-e22b-5df5-ae63-d93ebab69eaf"
 Statistics = "10745b16-79ce-11e8-11f9-7d13ad32a3b2"
 
 [compat]
+DataFrames = "1.8.0"
 DrWatson = "2.19.1"
 DuckDB = "1.3.2"
+FunSQL = "0.15.0"
 HuggingFaceHub = "0.1.2"
 JSON3 = "1.14.3"
 LibPQ = "1.18.0"

test/FunSQLTest.jl

@@ -2,8 +2,11 @@ import HuggingFaceHub as HF
 using DuckDB
 using JSON3
 using Test
+using DataFrames
+using FunSQL
 
-# Get the dataset object by id
+
+# Download dataset files
 synthea_ds = HF.info(HF.Dataset, "JuliaHealthOrg/JuliaHealthDatasets")
 funsql_ds = HF.info(HF.Dataset, "JuliaHealthOrg/FunSQLQueries")
 
@@ -21,18 +24,23 @@ data = JSON3.read(read(funsql_dataset_path, String))
         sql_query = row["sql_query"]
         funsql_code = row["response"]
 
-        # Evaluate FunSQL code to get the query object
-        funsql_query = eval(Meta.parse(funsql_code))
+        try
+            # Evaluate FunSQL code to get the query object
+            funsql_query = eval(Meta.parse(funsql_code))
 
-        # Render FunSQL query to SQL for DuckDB
-        funsql_sql = FunSQL.render(funsql_query, dialect=FunSQL.SQLDialect(:duckdb))
+            # Render FunSQL query to SQL for DuckDB
+            funsql_sql = FunSQL.render(funsql_query, dialect=FunSQL.SQLDialect(:duckdb))
 
-        # Execute both queries
-        sql_result = DuckDB.execute(conn, sql_query) |> DataFrame
-        funsql_result = DuckDB.execute(conn, funsql_sql) |> DataFrame
+            # Execute both queries
+            sql_result = DuckDB.execute(conn, sql_query) |> DataFrame
+            funsql_result = DuckDB.execute(conn, funsql_sql) |> DataFrame
 
-        # Compare results
-        is_equal = sql_result == funsql_result
+            # Compare results
+            is_equal = isequal(sql_result, funsql_result)
+        catch err
+            @warn "Query $i failed: $err"
+            is_equal = false
+        end
 
         @testset "Query $i" begin
             @test is_equal