Python clear_db_ids Examples

Example #1

File: base_state_entity_matcher_test.py Project: Alex-sConjecture/pulse-data

    def _assert_people_match(
            self, expected_people, matched_people, debug=False):
        converted_matched = \
            converter.convert_schema_objects_to_entity(matched_people)
        db_expected_with_backedges = \
            converter.convert_entity_people_to_schema_people(expected_people)
        expected_with_backedges = \
            converter.convert_schema_objects_to_entity(
                db_expected_with_backedges)

        clear_db_ids(converted_matched)
        clear_db_ids(expected_with_backedges)

        if debug:
            print('============== EXPECTED WITH BACKEDGES ==============')
            print_entity_trees(expected_with_backedges)
            print('============== CONVERTED MATCHED ==============')
            print_entity_trees(converted_matched)
        self.assertCountEqual(expected_with_backedges, converted_matched)

Example #2

File: base_direct_ingest_controller_tests.py Project: jazzPouls/pulse-data

 def convert_and_clear_db_ids(db_entities: List[StateBase]) -> List[Entity]:
     converted = StateSchemaToEntityConverter().convert_all(
         db_entities, populate_back_edges=True)
     clear_db_ids(converted)
     return converted