Add schema check in test

splitgraph · Jun 3, 2024 · 7a2dc3c · 7a2dc3c
1 parent 467dd48
commit 7a2dc3c
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/tests/basic_integration.rs b/tests/basic_integration.rs
@@ -2,6 +2,7 @@ use arrow::array::{
     Array, BinaryArray, BooleanArray, Date32Array, Decimal128Array, Float32Array, Float64Array,
     Int16Array, Int32Array, Int64Array, Int8Array, StringArray, TimestampMicrosecondArray,
 };
+use arrow::datatypes::DataType;
 use clap::Parser;
 use futures::{StreamExt, TryStreamExt};
 use lakehouse_loader::pg_arrow_source::PgArrowSource;
@@ -207,6 +208,12 @@ async fn test_pg_arrow_source() {
     assert!(!cdate_array.is_null(2));
     assert_eq!(cdate_array.value(2), elapsed_days as i32 + 2);
 
+    // THEN the numeric field data type should be as expected
+    assert_eq!(
+        *rb1.schema().field(11).data_type(),
+        DataType::Decimal128(8, 3)
+    );
+
     // THEN the first few numeric values should be as expected
     let cnumeric_array = rb1
         .column(11)