| 359 | } |
| 360 | |
| 361 | void AssertDatumsApproxEqual(const Datum& expected, const Datum& actual, bool verbose, |
| 362 | const EqualOptions& options) { |
| 363 | ASSERT_EQ(expected.kind(), actual.kind()) |
| 364 | << "expected:" << expected.ToString() << " got:" << actual.ToString(); |
| 365 | |
| 366 | switch (expected.kind()) { |
| 367 | case Datum::SCALAR: |
| 368 | AssertScalarsApproxEqual(*expected.scalar(), *actual.scalar(), verbose, options); |
| 369 | break; |
| 370 | case Datum::ARRAY: { |
| 371 | auto expected_array = expected.make_array(); |
| 372 | auto actual_array = actual.make_array(); |
| 373 | AssertArraysApproxEqual(*expected_array, *actual_array, verbose, options); |
| 374 | break; |
| 375 | } |
| 376 | case Datum::CHUNKED_ARRAY: { |
| 377 | auto expected_array = expected.chunked_array(); |
| 378 | auto actual_array = actual.chunked_array(); |
| 379 | AssertChunkedApproxEquivalent(*expected_array, *actual_array, options); |
| 380 | break; |
| 381 | } |
| 382 | default: |
| 383 | // TODO: Implement better print |
| 384 | ASSERT_TRUE(actual.Equals(expected)); |
| 385 | break; |
| 386 | } |
| 387 | } |
| 388 | |
| 389 | std::shared_ptr<Array> ArrayFromJSON(const std::shared_ptr<DataType>& type, |
| 390 | std::string_view json) { |
no test coverage detected