def test_parse_and_migrate_with_existing_data_should_do_nothing(): depotwerte = Depotwerte() raw_table = pd.DataFrame([["demoname", "demoisin", Depotwerte.TYP_FOND]], columns=Depotwerte.TABLE_HEADER) depotwerte.parse_and_migrate(raw_table) assert check_equal(depotwerte.content.columns, Depotwerte.TABLE_HEADER) assert len(depotwerte.content) == 1 assert depotwerte.content.Name[0] == 'demoname' assert depotwerte.content.ISIN[0] == 'demoisin' assert depotwerte.content.Typ[0] == Depotwerte.TYP_FOND
def test_parse_and_migrate_without_type_shoud_set_type_etf(): depotwerte = Depotwerte() raw_table = pd.DataFrame([["demoname", "demoisin"]], columns=['Name', 'ISIN']) depotwerte.parse_and_migrate(raw_table) assert check_equal(depotwerte.content.columns, Depotwerte.TABLE_HEADER) assert len(depotwerte.content) == 1 assert depotwerte.content.Name[0] == 'demoname' assert depotwerte.content.ISIN[0] == 'demoisin' assert depotwerte.content.Typ[0] == Depotwerte.TYP_ETF