update test with parquet save