Be more specific in KB serialization test.

This commit is contained in:
Raphael Mitsch 2023-02-23 15:14:18 +01:00
parent d8ac767e25
commit b6d9f4c5ab

View File

@ -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