From 8584082bd0b2ec88444eba059f435e41d9f225ba Mon Sep 17 00:00:00 2001 From: Benjamin Theunissen Date: Fri, 15 Mar 2024 17:26:11 -0400 Subject: [PATCH] Fix lint --- tests/target_test_cases.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/tests/target_test_cases.py b/tests/target_test_cases.py index 5882001..167f367 100644 --- a/tests/target_test_cases.py +++ b/tests/target_test_cases.py @@ -21,14 +21,14 @@ def validate(self) -> None: statement=text("SELECT * FROM date_type"), ).fetchall() record_id_1 = 1 - record_1 = [ + record_1 = next(iter([ record for record in result if record[0] == record_id_1 - ][0] + ])) assert record_1[1] == datetime.date(2024, 3, 15) record_id_2 = 2 - record_2 = [ + record_2 = next(iter([ record for record in result if record[0] == record_id_2 - ][0] + ])) assert record_2[1] == datetime.date(2024, 3, 16) custom_target_test_suite = TestSuite(