def test_nested_and_query_entity_equality(): """Tests NestedEntity and QueryEntity equality operations""" entity_a = NestedEntity( ("Entity", "Entity", "entity"), (Span(0, 5), ) * 3, (Span(0, 0), ) * 3, Entity("text", "type", "role", "value", "display"), ) entity_b = QueryEntity( ("Entity", "Entity", "entity"), (Span(0, 5), ) * 3, (Span(0, 0), ) * 3, Entity("text", "type", "role", "value", "display"), ) assert entity_a == entity_b assert not entity_a != entity_b entity_a = NestedEntity( ("Entity123", "Entity", "entity"), (Span(0, 5), ) * 3, (Span(0, 0), ) * 3, Entity("text", "type", "role", "value", "display"), ) entity_b = QueryEntity( ("Entity", "Entity", "entity"), (Span(0, 5), ) * 3, (Span(0, 0), ) * 3, Entity("text", "type", "role", "value", "display"), ) assert not entity_a == entity_b assert entity_a != entity_b
def test_query_entity_equality(): """Tests query entity equality""" entity_a = QueryEntity(('Entity', 'Entity', 'entity'), (Span(0, 5), ) * 3, (Span(0, 0), ) * 3, Entity('text', 'type', 'role', 'value', 'display')) entity_b = QueryEntity(('Entity', 'Entity', 'entity'), (Span(0, 5), ) * 3, (Span(0, 0), ) * 3, Entity('text', 'type', 'role', 'value', 'display')) assert entity_a == entity_b
def test_query_entity_equality(): """Tests query entity equality""" entity_a = QueryEntity( ("Entity", "Entity", "entity"), (Span(0, 5), ) * 3, (Span(0, 0), ) * 3, Entity("text", "type", "role", "value", "display"), ) entity_b = QueryEntity( ("Entity", "Entity", "entity"), (Span(0, 5), ) * 3, (Span(0, 0), ) * 3, Entity("text", "type", "role", "value", "display"), ) assert entity_a == entity_b
def test_nested_and_query_entity_equality(): """Tests NestedEntity and QueryEntity equality operations""" entity_a = NestedEntity(('Entity', 'Entity', 'entity'), (Span(0, 5), ) * 3, (Span(0, 0), ) * 3, Entity('text', 'type', 'role', 'value', 'display')) entity_b = QueryEntity(('Entity', 'Entity', 'entity'), (Span(0, 5), ) * 3, (Span(0, 0), ) * 3, Entity('text', 'type', 'role', 'value', 'display')) assert entity_a == entity_b assert not entity_a != entity_b entity_a = NestedEntity(('Entity123', 'Entity', 'entity'), (Span(0, 5), ) * 3, (Span(0, 0), ) * 3, Entity('text', 'type', 'role', 'value', 'display')) entity_b = QueryEntity(('Entity', 'Entity', 'entity'), (Span(0, 5), ) * 3, (Span(0, 0), ) * 3, Entity('text', 'type', 'role', 'value', 'display')) assert not entity_a == entity_b assert entity_a != entity_b