Skip to content

Commit cb11f72

Browse files
authored
[AL-6726] fix assert message (#1181)
2 parents f055052 + 41107cc commit cb11f72

File tree

1 file changed

+4
-2
lines changed

1 file changed

+4
-2
lines changed

tests/integration/test_data_rows.py

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -799,7 +799,9 @@ def test_data_row_bulk_creation_with_same_global_keys(dataset, sample_image,
799799
task.wait_till_done()
800800
if is_adv_enabled:
801801
assert task.status == "COMPLETE"
802+
assert type(task.failed_data_rows) is list
802803
assert len(task.failed_data_rows) == 1
804+
assert type(task.created_data_rows) is list
803805
assert len(task.created_data_rows) == 1
804806
assert task.failed_data_rows[0][
805807
'message'] == f"Duplicate global key: '{global_key_1}'"
@@ -859,8 +861,8 @@ def test_data_row_delete_and_create_with_same_global_key(
859861
if is_adv_enabled:
860862
assert task.status == "COMPLETE"
861863
assert len(task.failed_data_rows) == 1
862-
assert "Duplicate global key found" in task.failed_data_rows[0][
863-
'message']
864+
assert task.failed_data_rows[0][
865+
'message'] == f"Duplicate global key: '{global_key_1}'"
864866
else:
865867
assert task.status == "FAILED"
866868
assert len(task.failed_data_rows) > 0

0 commit comments

Comments
 (0)