From a6c0475e12970752df972b4659ca8a6c5c1c3b62 Mon Sep 17 00:00:00 2001 From: Jason Lowe Date: Tue, 20 Jul 2021 11:26:56 -0500 Subject: [PATCH] Assert that CPU and GPU row fields match when present Signed-off-by: Jason Lowe --- integration_tests/src/main/python/asserts.py | 1 + 1 file changed, 1 insertion(+) diff --git a/integration_tests/src/main/python/asserts.py b/integration_tests/src/main/python/asserts.py index f07c9cdf0a4..a08473ef5fb 100644 --- a/integration_tests/src/main/python/asserts.py +++ b/integration_tests/src/main/python/asserts.py @@ -30,6 +30,7 @@ def _assert_equal(cpu, gpu, float_check, path): if (t is Row): assert len(cpu) == len(gpu), "CPU and GPU row have different lengths at {} CPU: {} GPU: {}".format(path, len(cpu), len(gpu)) if hasattr(cpu, "__fields__") and hasattr(gpu, "__fields__"): + assert cpu.__fields__ == gpu.__fields__, "CPU and GPU row have different fields at {} CPU: {} GPU: {}".format(path, cpu.__fields__, gpu.__fields__) for field in cpu.__fields__: _assert_equal(cpu[field], gpu[field], float_check, path + [field]) else: