1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950 |
- '''
- Created on Jan 29, 2020 @author: deeejas
- '''
- import unittest
- import pandas as pd
- from winmentor import info_anaf
- from winmentor.dbread import WinMentor
- from winmentor.mentorutils import Mentorutils
- class Test(unittest.TestCase):
-
- mentor = WinMentor()
- utils = Mentorutils()
-
- @unittest.skip('reason')
- def test_check_list(self):
- info_anaf.check_list('lista_cf.txt')
-
- def test_utils(self):
- headers = ['Denumire', 'CF', 'J', 'Adresa', 'Oras', 'Judet', 'Prescurtat', 'Obs']
- n_headers = ['Denumire', 'CF', 'J', 'Adresa', 'Oras', 'Judet', 'Prescurtat', 'Obs',
- 'TurnO', 'Result', 'divid', 'divid_current', 'spons', 'month']
- mentor = WinMentor()
- utils = Mentorutils()
- office = mentor.filtered_firmlist('/firme.db', headers=headers)
- utils.gen_turnover('./generated_files/turn_over', 'to', office, m_month='2019_12', n_headers=n_headers)
-
- def test_get_bank_accounts(self):
- for account in self.mentor.get_bank_accounts('WEBS'):
- # if account[2].startswith(' '):
- print(account)
-
- def test_get_oblig(self):
- head = ['Denumire', 'CF', 'J', 'Adresa', 'Oras', 'Judet', 'Prescurtat', 'Admin', 'AdminP', 'RCNP', 'Obs']
- # firm_list = list(self.mentor.filtered_firmlist(head))
- df_dicts = self.mentor.get_oblig('KINVEST')
- df = pd.DataFrame(df_dicts[0])
- df2 = pd.DataFrame(df_dicts[1])
- df3 = pd.merge(df, df2, how='left', left_on='Part', right_on='Cod')
- df3.drop(columns=['Part', 'Cod'], inplace=True)
- print(df3.head(10))
- df3.to_excel('obligatii.xlsx')
-
- if __name__ == "__main__":
- unittest.main()
|