예제 #1
0
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
예제 #2
0
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