def test_dataframes(): # setup test data wipe_db_tables() # WARNING: kills existing database data labelled_rows = raw_to_labelled_rows(raw_rows=doc_as_iterable(INVESTMENT_DOC), spec_dicts=(REF_HEADER_DICT, REF_UNIT_DICT)) assert labelled_rows == INVESTMENT_PARSED flat_rows = list(stream_flat_data(labelled_rows)) assert flat_rows == INVESTMENT_FLAT_ROW stream_to_database(flat_rows) # check user-end output check_final_dataframes()
def test_dataframes(self): # setup test data wipe_db_tables() # WARNING: kills existing database data labelled_rows = raw_to_labelled_rows(raw_rows=self.doc_as_iterable(self.INVESTMENT_DOC), spec_dicts=(self.REF_HEADER_DICT, self.REF_UNIT_DICT)) self.assertEqual(labelled_rows, self.INVESTMENT_PARSED) flat_rows = list(stream_flat_data(labelled_rows)) self.assertEqual(flat_rows, self.INVESTMENT_FLAT_ROW) stream_to_database(flat_rows) # check user-end output self.check_final_dataframes()
def test_dataframes(self): # setup test data wipe_db_tables() # WARNING: kills existing database data labelled_rows = raw_to_labelled_rows( raw_rows=self.doc_as_iterable(self.INVESTMENT_DOC), spec_dicts=(self.REF_HEADER_DICT, self.REF_UNIT_DICT)) self.assertEqual(labelled_rows, self.INVESTMENT_PARSED) flat_rows = list(stream_flat_data(labelled_rows)) self.assertEqual(flat_rows, self.INVESTMENT_FLAT_ROW) stream_to_database(flat_rows) # check user-end output self.check_final_dataframes()
def test_dataframes(): # setup test data wipe_db_tables() # WARNING: kills existing database data labelled_rows = raw_to_labelled_rows( raw_rows=doc_as_iterable(INVESTMENT_DOC), spec_dicts=(REF_HEADER_DICT, REF_UNIT_DICT)) assert labelled_rows == INVESTMENT_PARSED flat_rows = list(stream_flat_data(labelled_rows)) assert flat_rows == INVESTMENT_FLAT_ROW stream_to_database(flat_rows) # check user-end output check_final_dataframes()