class XbrlIncomeStatementAssemblerTest(unittest.TestCase):
def setUp(self):
self.assembler = XbrlIncomeStatementAssembler()
self.file_utils = FileUtils()
def tearDown(self):
self.assembler = None
self.file_utils = None
def test_assemble_2330_in_2014Q3(self):
# online: http://mops.twse.com.tw/server-java/t164sb01?step=1&CO_ID=2330&SYEAR=2014&SSEASON=3&REPORT_ID=C
path = './stockcat/tests/unit/data/xbrl_financial_statement/2330/2014/03.html'
param = {
'content' : self.file_utils.read_file(path),
'stock_symbol' : '2330',
'date' : datetime.date(2014, 9, 30),
}
dao = self.assembler.assemble(param)
column_name_list = dao.get_column_name_list()
row_list = dao.get_row_list()
self.assertEqual(column_name_list[0], u'會計項目')
self.assertEqual(column_name_list[1], (datetime.date(2014, 1, 1), datetime.date(2014, 9, 30)))
self.assertEqual(column_name_list[2], (datetime.date(2013, 1, 1), datetime.date(2013, 9, 30)))
self.assertEqual(row_list[0], [u'營業收入'])
self.assertEqual(row_list[1], [u'營業收入合計', 209049734, 162577034, 540285390, 451218350])
class IncomeStatementAssembler():
def __init__(self):
self.xbrl_assembler = XbrlIncomeStatementAssembler()
self.legacy_assembler = LegacyIncomeStatementAssembler()
# IFRS are available after year 2013. Legacy are available before year 2013.
self.splitted_date = datetime.date(2013, 1, 1)
def assemble(self, param):
if param['date'] >= self.splitted_date:
return self.xbrl_assembler.assemble(param)
else:
return self.legacy_assembler.assemble(param)
def __init__(self):
self.xbrl_assembler = XbrlIncomeStatementAssembler()
self.legacy_assembler = LegacyIncomeStatementAssembler()
# IFRS are available after year 2013. Legacy are available before year 2013.
self.splitted_date = datetime.date(2013, 1, 1)
def setUp(self):
self.assembler = XbrlIncomeStatementAssembler()
self.file_utils = FileUtils()