Skip to content

Commit

Permalink
refactor; robustify test
Browse files Browse the repository at this point in the history
  • Loading branch information
doctrino committed Dec 12, 2024
1 parent 7cc8e07 commit d15ec32
Showing 1 changed file with 5 additions and 5 deletions.
10 changes: 5 additions & 5 deletions tests/tests_integration/test_api/test_sequences.py
Original file line number Diff line number Diff line change
@@ -1,4 +1,4 @@
from unittest import mock
from unittest import TestCase, mock

import pytest

Expand Down Expand Up @@ -386,7 +386,7 @@ def test_upsert_sequence_replace(self, cognite_client: CogniteClient) -> None:
retrieved = cognite_client.sequences.retrieve(external_id=upserted.external_id)

assert retrieved is not None
assert retrieved.as_write().columns.dump() == upsert.columns.dump()
TestCase().assertCountEqual(retrieved.as_write().columns.dump(), upsert.columns.dump())
finally:
if created:
cognite_client.sequences.delete(external_id=created.external_id, ignore_unknown_ids=True)
Expand Down Expand Up @@ -430,9 +430,9 @@ def test_upsert_sequence_patch(self, cognite_client: CogniteClient) -> None:
retrieved = cognite_client.sequences.retrieve(external_id=upserted.external_id)

assert retrieved is not None
assert (
retrieved.as_write().columns.dump()
== SequenceColumnWriteList([original_sequence.columns[0], upsert.columns[0]]).dump()
TestCase().assertCountEqual(
retrieved.as_write().columns.dump(),
SequenceColumnWriteList([original_sequence.columns[0], upsert.columns[0]]).dump(),
)
finally:
if created:
Expand Down

0 comments on commit d15ec32

Please sign in to comment.