def test_claimed_recids_with_missing_data(): from beard_server.modules.clustering.recid import make_recid_clusters signatures = [{'signature_id': 'ABC', 'author_claimed': True}, {'author_recid': '1', 'author_claimed': True}] output = {} assert make_recid_clusters(signatures, claimed_only=True) == output
def test_missing_recids(): """Test if signatures with no recids will be skipped.""" from beard_server.modules.clustering.recid import make_recid_clusters signatures = [{'signature_id': 'ABC', 'author_recid': '1'}, {'signature_id': 'XYZ', 'author_recid': False}] output = {'1': ['ABC']} assert make_recid_clusters(signatures, claimed_only=False) == output
def test_broken_signatures(): """Test if signatures with no valid fields will be skipped.""" from beard_server.modules.clustering.recid import make_recid_clusters signatures = [{'uuid': 'ABC', 'recid': '1'}, {'signature_id': 'CDE'}, {'recid': '2'}] output = {} assert make_recid_clusters(signatures, claimed_only=False) == output
def test_recids(): """Test if the two exact clusters will be matched.""" from beard_server.modules.clustering.recid import make_recid_clusters signatures = [{'signature_id': 'ABC', 'author_recid': '1'}, {'signature_id': 'CDE', 'author_recid': '1'}, {'signature_id': 'XYZ', 'author_recid': '2'}] output = {'1': ['ABC', 'CDE'], '2': ['XYZ']} assert make_recid_clusters(signatures, claimed_only=False) == output
def test_claimed_recids(): """Test if claimed signatures will be clustered together.""" from beard_server.modules.clustering.recid import make_recid_clusters signatures = [{'signature_id': 'ABC', 'author_recid': '1', 'author_claimed': True}, {'signature_id': 'CDE', 'author_recid': '1', 'author_claimed': True}, {'signature_id': 'FGH', 'author_recid': '2', 'author_claimed': True}, {'signature_id': 'XYZ', 'author_recid': '2', 'author_claimed': False}] output = {'1': ['ABC', 'CDE'], '2': ['FGH']} assert make_recid_clusters(signatures, claimed_only=True) == output