def test_remove_id_in_single_relationship(): relationship_store = RelationshipStore(relationships) relationship_store.remove_all_relationships_for_list_item_id("789101") remaining_relationship = Relationship(**relationships[1]) assert len(relationship_store) == 1 assert (relationship_store.get_relationship( remaining_relationship.list_item_id, remaining_relationship.to_list_item_id) == remaining_relationship) assert relationship_store.is_dirty
def test_remove_id_in_multiple_relationships(): relationship_store = RelationshipStore(relationships) relationship_store.remove_all_relationships_for_list_item_id("123456") assert not relationship_store assert relationship_store.is_dirty