adding csv test
This commit is contained in:
parent
f6c083ac43
commit
e61ab45990
@ -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)))
|
||||
|
23
test/test_csv.ml
Normal file
23
test/test_csv.ml
Normal file
@ -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
|
||||
]
|
||||
]
|
Loading…
Reference in New Issue
Block a user