From e61ab45990695dcc9e648d24df50da57245e0953 Mon Sep 17 00:00:00 2001 From: simon petit Date: Mon, 5 May 2025 15:35:49 +0000 Subject: [PATCH] adding csv test --- test/dune | 8 +++++++- test/test_csv.ml | 23 +++++++++++++++++++++++ 2 files changed, 30 insertions(+), 1 deletion(-) create mode 100644 test/test_csv.ml diff --git a/test/dune b/test/dune index 9791cc6..f00f6a0 100644 --- a/test/dune +++ b/test/dune @@ -6,8 +6,14 @@ (name test_logical_plan) (libraries ast logical_plan alcotest)) +(test + (name test_csv) + (libraries csv ast logical_plan alcotest) + (deps + (file ./files/csv/data.csv))) + (test (name test_parquet) (libraries parquet) (deps - (file ./files/parquet/mock_data.parquet))) + (file ./files/parquet/data.parquet))) diff --git a/test/test_csv.ml b/test/test_csv.ml new file mode 100644 index 0000000..be3ebdb --- /dev/null +++ b/test/test_csv.ml @@ -0,0 +1,23 @@ +open Csv +open Alcotest + +(* Function to read CSV file *) +let read_csv filename = + let ic = open_in filename in + let csv = Csv.load_in ic in + csv + +(* Test to check if the CSV file is read correctly *) +let test_read_csv () = + let csv_data = read_csv "files/csv/data.csv" in + let nb_columns = Csv.columns csv_data in + check int "Number of columns" nb_columns 5 + + +(* Run the test *) +let () = + run "CSV Tests" [ + "Read CSV", [ + test_case "Read and validate CSV" `Quick test_read_csv + ] + ]