Skip to content

Commit

Permalink
test: add test for metadata conversion with None simprints
Browse files Browse the repository at this point in the history
  • Loading branch information
titusz committed Aug 13, 2024
1 parent f71009b commit 2d14293
Showing 1 changed file with 10 additions and 0 deletions.
10 changes: 10 additions & 0 deletions tests/test_models.py
Original file line number Diff line number Diff line change
Expand Up @@ -84,6 +84,16 @@ def test_metadata_to_object_format():
assert object_meta2.model_dump() == object_meta.model_dump()


def test_metadata_to_index_format_with_none_simprints():
# Test conversion when feature_set.simprints is None
features = [FeatureSet(simprints=None, embedding=[0.1, 0.2])]
meta = Metadata(iscc="ISCC1234567890", features=features)
index_meta = meta.to_index_format()
assert index_meta.features[0].simprints is None
assert index_meta.features[0].embedding == [0.1, 0.2]
assert index_meta.model_dump() == meta.model_dump()


def test_metadata_format_conversion_with_no_features():
meta = Metadata(iscc="ISCC1234567890")
index_meta = meta.to_index_format()
Expand Down

0 comments on commit 2d14293

Please sign in to comment.