def test_stock_only_files(self):
"""
Test ready stock only files
"""
test_date = '2014-03-10'
test_path = FILES['tos_positions'] + 'tests/' + '2014-03-10-stock.csv'
print 'file date: %s' % test_date
print 'file path: %s\n' % test_path
open_csv = OpenPosCSV(fname=test_path)
positions, overalls = open_csv.read()
for position in positions:
for key, item in position.items():
if key == 'Symbol' or key == 'Company':
self.assertEqual(type(item), str)
elif key == 'Instrument' or key == 'Stock':
self.assertEqual(type(item), dict)
self.assertIn(len(item), (0, 14))
elif key == 'Options':
self.assertEqual(type(item), list)
for key in ['Symbol', 'Company', 'Instrument', 'Stock', 'Options']:
self.assertIn(key, position.keys())
print '%s found in positions!' % key
print ''
def setUp(self):
print '=' * 100
print "%s: currently run: %s" % (self.__class__.__name__,
self._testMethodName)
print '-' * 100 + '\n'
test_file = FILES['tos_positions'] + r'2014-08-01-PositionStatement.csv'
self.open_csv = OpenPosCSV(fname=test_file)
def test_2014_09_06_file(self):
f = FILES['tos_positions'] + '2014-09-06-PositionStatement.csv'
open_csv = OpenPosCSV(fname=f)
pos, ov = open_csv.read()
print 'fname: %s' % f
print 'positions length: %d' % len(pos)
print 'overall length: %d\n' % len(ov)
def test_ready_multiple_files(self):
"""
Test read on all files inside tos.pos
"""
files = [
'2014-03-07-closed.csv', '2014-03-10-stock.csv',
'2014-03-11-hedge.csv', '2014-03-12-one-leg.csv',
'2014-03-13-two-legs.csv', '2014-03-14-three-legs.csv',
'2014-03-17-four-legs-part-1.csv'
]
for f in files:
open_csv = OpenPosCSV(fname=FILES['tos_positions'] + 'tests/' + f)
pos, ov = open_csv.read()
print 'fname: %s' % f
print 'positions length: %d' % len(pos)
print 'overall length: %d\n' % len(ov)
self.assertGreater(len(pos), 1)
self.assertEqual(len(ov), 5)
