From 50ba7a099e33a5bf70da1e5cf6c4a1104e1c4419 Mon Sep 17 00:00:00 2001 From: Marcos Martinez Date: Wed, 6 Sep 2023 10:35:20 +0100 Subject: [PATCH] :white_check_mark: Updated tests Signed-off-by: Marcos Martinez --- zshot/tests/utils/test_data_models.py | 12 ++++++------ 1 file changed, 6 insertions(+), 6 deletions(-) diff --git a/zshot/tests/utils/test_data_models.py b/zshot/tests/utils/test_data_models.py index a530611..885ef29 100644 --- a/zshot/tests/utils/test_data_models.py +++ b/zshot/tests/utils/test_data_models.py @@ -85,18 +85,18 @@ def test_relation_span(): s2 = Span.from_dict(EX_DATASET_RELATIONS['sentence_entities'][0][1]) rs = RelationSpan(start=s1, end=s2, relation=Relation(name=EX_DATASET_RELATIONS['labels'][0]), score=1, kb_id='P1') assert type(rs) is RelationSpan - assert type(rs.start) == Span - assert type(rs.end) == Span - assert type(rs.relation) == Relation + assert type(rs.start) is Span + assert type(rs.end) is Span + assert type(rs.relation) is Relation assert rs.score == 1 assert rs.kb_id == 'P1' # No Score/KB Id rs = RelationSpan(start=s1, end=s2, relation=Relation(name=EX_DATASET_RELATIONS['labels'][0])) assert type(rs) is RelationSpan - assert type(rs.start) == Span - assert type(rs.end) == Span - assert type(rs.relation) == Relation + assert type(rs.start) is Span + assert type(rs.end) is Span + assert type(rs.relation) is Relation # Check hash assert hash(rs) == 1864423560