class StockCapitalIncreaseHistoryAssemblerTest(unittest.TestCase):
def setUp(self):
self.assembler = CapitalIncreaseHistoryAssembler()
self.file_utils = FileUtils()
def tearDown(self):
self.assembler = None
self.file_utils = None
def test_assemble_2498(self):
# online: http://jdata.yuanta.com.tw/z/zc/zcb/zcb_2498.djhtm
path = './stockcat/tests/unit/data/capital_increase_history/2498.html'
param = {
'content' : self.file_utils.read_file(path),
'stock_symbol' : '2498'
}
dao = self.assembler.assemble(param)
self.assertEqual(dao.get_stock_symbol(), param['stock_symbol'])
actual = dao.get_column_name_list()
expected = [u'年度', u'現金增資', u'比重', u'盈餘轉增資', u'比重', u'公積及其他', u'比重']
self.assertEqual(actual, expected)
row_list = dao.get_row_list()
self.assertEqual(row_list[0], [datetime.date(2015, 12, 31), 14.464519, 0.1747, 67.724067, 0.8180, 0.598634, 0.0072])
for row in row_list:
self.assertEqual(len(row), 7)
def test_assemble_raise_no_record_assemble_error(self):
# online: http://jdata.yuanta.com.tw/z/zc/zcb/zcb_3009.djhtm
path = './stockcat/tests/unit/data/error/capital_increase_history_not_found_error.html'
param = {
'content' : self.file_utils.read_file(path),
'stock_symbol' : '3009'
}
with self.assertRaises(NoRecordAssembleError) as context:
self.assembler.assemble(param)
self.assertEqual(context.exception.param['stock_symbol'], param['stock_symbol'])
def setUp(self):
self.assembler = CapitalIncreaseHistoryAssembler()
self.file_utils = FileUtils()