def test_add_shouldAdd(self): component_under_test = Depotwerte() component_under_test.add(name='1name', isin='some isin') assert len(component_under_test.content) == 1 assert component_under_test.content.Name[0] == '1name' assert component_under_test.content.ISIN[0] == 'some isin'
def test_add_shouldAdd(): component_under_test = Depotwerte() component_under_test.add(name='1name', isin='some isin', typ=component_under_test.TYP_ETF) assert len(component_under_test.content) == 1 assert component_under_test.content.Name[0] == '1name' assert component_under_test.content.ISIN[0] == 'some isin' assert component_under_test.content.Typ[0] == component_under_test.TYP_ETF
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
def __init__(self, name, ausgeschlossene_kategorien=set()): self.name = name self.dauerauftraege = Dauerauftraege() self.gemeinsamebuchungen = Gemeinsamebuchungen() self.einzelbuchungen = Einzelbuchungen() self.sparbuchungen = Sparbuchungen() self.sparkontos = Kontos() self.depotwerte = Depotwerte() self.order = Order() self.orderdauerauftrag = OrderDauerauftrag() self.depotauszuege = Depotauszuege() self.einzelbuchungen.ausgeschlossene_kategorien = ausgeschlossene_kategorien self.tainted = 0
def test_get_valid_isins(self): component_under_test = Depotwerte() component_under_test.add('invalid isin', '-') component_under_test.add('valid isin', 'isin56789012') assert component_under_test.get_valid_isins() == ['isin56789012']
def test_get_isin_nach_typ(): depotwerte = Depotwerte() depotwerte.add('Name_Etf1', 'ISIN_Etf1', depotwerte.TYP_ETF) depotwerte.add('Name_Etf2', 'ISIN_Etf2', depotwerte.TYP_ETF) depotwerte.add('Name_Fond', 'ISIN_Fond', depotwerte.TYP_FOND) result = depotwerte.get_isin_nach_typ() assert len(result.keys()) == 2 assert result[depotwerte.TYP_ETF] == ['ISIN_Etf1', 'ISIN_Etf2'] assert result[depotwerte.TYP_FOND] == ['ISIN_Fond']
def test_get_valid_isins(): component_under_test = Depotwerte() component_under_test.add(name='invalid isin', isin='-', typ=component_under_test.TYP_ETF) component_under_test.add(name='valid isin', isin='isin56789012', typ=component_under_test.TYP_ETF) assert component_under_test.get_valid_isins() == ['isin56789012']
def test_get_depotwerte(): component_under_test = Depotwerte() component_under_test.add(name='0name', isin='0isin', typ=component_under_test.TYP_ETF) component_under_test.add(name='1name', isin='1isin', typ=component_under_test.TYP_ETF) assert component_under_test.get_depotwerte() == ['0isin', '1isin']
def test_get_depotwerte(self): component_under_test = Depotwerte() component_under_test.add( '0name', '0isin', ) component_under_test.add( '1name', '1isin', ) assert component_under_test.get_depotwerte() == ['0isin', '1isin']
def test_get_isin_nach_typ_with_empty_db_should_return_empty(): depotwerte = Depotwerte() assert not depotwerte.get_isin_nach_typ()
class Database: def __init__(self, name, ausgeschlossene_kategorien=set()): self.name = name self.dauerauftraege = Dauerauftraege() self.gemeinsamebuchungen = Gemeinsamebuchungen() self.einzelbuchungen = Einzelbuchungen() self.sparbuchungen = Sparbuchungen() self.sparkontos = Kontos() self.depotwerte = Depotwerte() self.order = Order() self.orderdauerauftrag = OrderDauerauftrag() self.depotauszuege = Depotauszuege() self.einzelbuchungen.ausgeschlossene_kategorien = ausgeschlossene_kategorien self.tainted = 0 def taint(self): self.tainted = self.tainted + 1 def is_tainted(self): return self.taint_number() != 0 def taint_number(self): return self.tainted + \ self.dauerauftraege.taint_number() + \ self.einzelbuchungen.taint_number() + \ self.gemeinsamebuchungen.taint_number() + \ self.sparbuchungen.taint_number() + \ self.sparkontos.taint_number() + \ self.depotwerte.taint_number() + \ self.order.taint_number() +\ self.depotauszuege.taint_number() +\ self.orderdauerauftrag.taint_number() def refresh(self): print('DATABASE: Erneuere Datenbestand') alle_dauerauftragsbuchungen = self.dauerauftraege.get_all_einzelbuchungen_until_today( ) self.einzelbuchungen.append_row(alle_dauerauftragsbuchungen) anteil_gemeinsamer_buchungen = self.gemeinsamebuchungen.anteil_gemeinsamer_buchungen( ) self.einzelbuchungen.append_row(anteil_gemeinsamer_buchungen) anteil_sparbuchungen = self.sparbuchungen.get_dynamische_einzelbuchungen( ) self.einzelbuchungen.append_row(anteil_sparbuchungen) anteil_orderdauerauftrag = self.orderdauerauftrag.get_all_order_until_today( ) self.order.append_row(anteil_orderdauerauftrag) anteil_order = self.order.get_dynamische_einzelbuchungen() self.einzelbuchungen.append_row(anteil_order) print('DATABASE: Datenbestand erneuert') def _write_trenner(self, abrechnunsdatei): return abrechnunsdatei.write("".rjust(40, "#") + "\n ") def abrechnen(self, mindate, maxdate, set_ergebnis=None, verhaeltnis=50, set_self_kategorie=None, set_other_kategorie=None): selector = self.gemeinsamebuchungen.select().select_range( mindate, maxdate) name_self = persisted_state.database_instance().name name_partner = viewcore.name_of_partner() gemeinsame_buchungen_content = selector.content select_partner = selector.fuer(name_partner) select_self = selector.fuer(name_self) summe_partner = select_partner.sum() summe_self = select_self.sum() ausgaben_gesamt = selector.sum() abrechnunsdatei = StringWriter() zeitraum = datum_to_german(mindate) + '-' + datum_to_german(maxdate) abrechnunsdatei.write_line('Abrechnung vom ' + datum_to_german(time.today()) + ' (' + zeitraum + ')') self._write_trenner(abrechnunsdatei) abrechnunsdatei.write_line('Ergebnis:') abrechnunsdatei.write_line(set_ergebnis) abrechnunsdatei.write_empty_line() self._write_large_table_row(abrechnunsdatei, 'Ausgaben von ' + name_partner, summe_partner) self._write_large_table_row(abrechnunsdatei, 'Ausgaben von ' + name_self, summe_self) abrechnunsdatei.write_line("".ljust(38, "-")) self._write_large_table_row(abrechnunsdatei, "Gesamt", ausgaben_gesamt) if verhaeltnis == 50: self.write_into_file(abrechnunsdatei, selector.faktor(0.5).to_list(), 'Gesamtausgaben pro Person ') self.write_into_file(abrechnunsdatei, select_partner.to_list(), 'Ausgaben von ' + name_partner) self.write_into_file(abrechnunsdatei, select_self.to_list(), 'Ausgaben von ' + name_self) ausgaben_fuer_partner = DataFrame() faktor_partner = self._faktor_other(verhaeltnis) ausgaben_fuer_self = DataFrame() faktor_self = self._faktor_self(verhaeltnis) summe_halb = selector.faktor(0.5).sum() if set_self_kategorie: faktor_self = 0.5 if set_other_kategorie: faktor_partner = 0.5 for _, row in gemeinsame_buchungen_content.iterrows(): buchung_partner = self._berechne_abbuchung( row['Datum'], row['Kategorie'], row['Name'], ("%.2f" % (row['Wert'] * faktor_partner))) buchung_partner.Dynamisch = False ausgaben_fuer_partner = pd.concat( [ausgaben_fuer_partner, buchung_partner]) buchung_self = self._berechne_abbuchung( row['Datum'], row['Kategorie'], row['Name'], ("%.2f" % (row['Wert'] * faktor_self))) buchung_self.Dynamisch = False ausgaben_fuer_self = pd.concat([ausgaben_fuer_self, buchung_self]) if set_self_kategorie: extra_wert = (ausgaben_gesamt * self._faktor_self(verhaeltnis)) - summe_halb extra_ausgleichs_buchung = self._berechne_abbuchung( maxdate, set_self_kategorie, set_self_kategorie, ("%.2f" % extra_wert)) extra_ausgleichs_buchung.Dynamisch = False ausgaben_fuer_self = pd.concat( [ausgaben_fuer_self, extra_ausgleichs_buchung]) if set_other_kategorie: extra_wert = (ausgaben_gesamt * self._faktor_other(verhaeltnis)) - summe_halb extra_ausgleichs_buchung = self._berechne_abbuchung( maxdate, set_other_kategorie, set_other_kategorie, ("%.2f" % extra_wert)) extra_ausgleichs_buchung.Dynamisch = False ausgaben_fuer_partner = pd.concat( [ausgaben_fuer_partner, extra_ausgleichs_buchung]) report = TextReportWriter().generate_report( ausgaben_fuer_partner, abrechnunsdatei.to_string()) self.einzelbuchungen.append_row(ausgaben_fuer_self) self.einzelbuchungen.taint() self.gemeinsamebuchungen.drop( gemeinsame_buchungen_content.index.tolist()) self.taint() write_abrechnung("Abrechnung_" + str(time.now()), report) return report def _faktor_self(self, verhaeltnis): return verhaeltnis / 100 def _faktor_other(self, verhaeltnis): return self._faktor_self(100 - verhaeltnis) def _write_large_table_row(self, abrechnunsdatei, name, summe): abrechnunsdatei.write_line( name.ljust(30, " ") + str("%.2f" % summe).rjust(7, " ")) def write_into_file(self, abrechnunsdatei, ausgaben, title): abrechnunsdatei.write_empty_line(count=2) self._write_trenner(abrechnunsdatei) abrechnunsdatei.write_line(title) self._write_trenner(abrechnunsdatei) self._write_tabelle(abrechnunsdatei, ausgaben) def _write_tabelle(self, writer, tabelle): writer.write_line( self._to_left("Datum", 10) + self._to_left(" Kategorie", 14) + self._to_left("Name", 21) + self._to_right("Wert", 7)) for row in tabelle: writer.write_line( datum_to_german(row['Datum']) + " " + row['Kategorie'].ljust(len("Kategorie "), " ") + " " + row['Name'].ljust(20, " ") + " " + str("%.2f" % (row['Wert'])).rjust(7, " ")) def _to_left(self, target_string, size): return target_string.ljust(size, ' ') def _to_right(self, target_string, size): return target_string.rjust(size, ' ') def _berechne_abbuchung(self, laufdatum, kategorie, name, wert): return DataFrame([[laufdatum, kategorie, name, wert, True]], columns=('Datum', 'Kategorie', 'Name', 'Wert', 'Dynamisch')) def _row_to_dict(self, columns, index, row_data): row = {'index': index} for key in columns: row[key] = row_data[key] return row def frame_to_list_of_dicts(self, dataframe): result_list = [] for index, row_data in dataframe.iterrows(): row = self._row_to_dict(dataframe.columns, index, row_data) result_list.append(row) return result_list
def test_edit_shouldEdit(self): component_under_test = Depotwerte() component_under_test.add( '0name', '0isin', ) component_under_test.add( '1name', '1isin', ) component_under_test.add( '2name', '2isin', ) assert len(component_under_test.content) == 3 element_before = component_under_test.get(1) assert element_before == {'index': 1, 'Name': '1name', 'ISIN': '1isin'} component_under_test.edit(index=1, name='13name', isin='13isin') assert len(component_under_test.content) == 3 element_after = component_under_test.get(1) assert element_after == { 'index': 1, 'Name': '13name', 'ISIN': '13isin' }