mirror of
https://github.com/explosion/spaCy.git
synced 2025-07-10 16:22:29 +03:00
Be more specific in KB serialization test.
This commit is contained in:
parent
d8ac767e25
commit
b6d9f4c5ab
|
@ -161,4 +161,4 @@ def test_serialize_subclassed_kb():
|
|||
# After IO, the KB is the standard one
|
||||
assert type(entity_linker2.kb) == SubInMemoryLookupKB
|
||||
assert entity_linker2.kb.entity_vector_length == 342
|
||||
assert hasattr(entity_linker2.kb, "custom_field")
|
||||
assert entity_linker2.kb.custom_field == 666
|
||||
|
|
Loading…
Reference in New Issue
Block a user