Method equals_reader

arrow-integration-test/src/lib.rs:221–242

Compare the Arrow JSON with a record batch reader

(&self, reader: &mut dyn RecordBatchReader)

impl ArrowJson {
    /// Compare the Arrow JSON with a record batch reader
    pub fn equals_reader(&self, reader: &mut dyn RecordBatchReader) -> Result<bool> {
        if !self.schema.equals_schema(&reader.schema()) {
            return Ok(false);
        }

        for json_batch in self.get_record_batches()?.into_iter() {
            let batch = reader.next();
            match batch {
                Some(Ok(batch)) => {
                    if json_batch != batch {
                        println!("json: {json_batch:?}");
                        println!("batch: {batch:?}");
                        return Ok(false);
                    }
                }
                Some(Err(e)) => return Err(e),
                None => return Ok(false),
            }
        }

        Ok(true)
    }

    /// Convert the stored dictionaries to `Vec[RecordBatch]`
    pub fn get_record_batches(&self) -> Result<Vec<RecordBatch>> {