示例#1
0
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
示例#2
0
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
示例#3
0
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
示例#4
0
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