def test_csv_file_import_to_StockHistory(self):
stock = StockSymbolFactory.create(name='ASHOK Leyland',
symbol='ASHOKLEY',
tickerNumber='500477')
stock.save()
self.assertEquals(StockSymbol.objects.count(), 1)
csv_filename = ut.get_csv_filename(stock)
try:
with transaction.atomic():
count = ut.read_csv_file(stock, csv_filename)
except:
self.fail("exception when trying to insert")
#count = StockHistory.objects.count()
self.assertGreater(StockHistory.objects.count(), 1)
lastRecord = StockHistory.objects.all().last()
self.assertEquals(lastRecord.openPrice, 50.5)
self.assertEquals(lastRecord.highPrice, 52.6)
self.assertEquals(lastRecord.lowPrice, 50.0)
self.assertEquals(lastRecord.closePrice, 52.3)
self.assertEquals(lastRecord.wap, 51.723125410140653536)
self.assertEquals(lastRecord.numberOfShares, 3035544)
self.assertEquals(lastRecord.numberOfTrades, 8128)
self.assertEquals(lastRecord.totalTurnover, 157007823)
self.assertEquals(lastRecord.spreadHighLow, 2.60)
self.assertEquals(lastRecord.spreadCloseOpen, 1.80)
def test_derived_csv_filename_is_found(self):
stock = StockSymbolFactory.create(name='ASHOK Leyland',
symbol='ASHOKLEY',
tickerNumber='500477')
csv_filename = ut.get_csv_filename(stock)
csv_file = ut.read_csv_file(stock, csv_filename)
self.assertNotEquals(csv_file, False)
def test_csv_file_import_to_StockHistory(self):
stock = StockSymbolFactory.create(name='ASHOK Leyland',
symbol='ASHOKLEY', tickerNumber='500477')
csv_filename = ut.get_csv_filename(stock)
try:
with transaction.atomic():
ut.read_csv_file(stock, csv_filename)
self.assertGreaterEqual(StockHistory.objects.count(), 1)
print(StockHistory.objects.count())
except:
pass
def test_derived_csv_filename_is_found_and_opened(self):
stock = StockSymbolFactory.create(name='ASHOK Leyland',
symbol='ASHOKLEY',
tickerNumber='500477')
csv_filename = ut.get_csv_filename(stock)
try:
file = open(csv_filename, 'r')
except Exception as e:
template = "An exception of type {0} occurred. Arguments:\n{1!r}"
message = template.format(type(e).__name__, e.args)
self.fail(message)
def test_derived_csv_filename_is_correct(self):
stock = StockSymbolFactory.create(name='ASHOK Leyland',
symbol='ASHOKLEY',
tickerNumber='500477')
csv_filename = ut.get_csv_filename(stock)
# real_filename = os.path.dirname(os.path.realpath(__file__))
real_filename = os.getcwd()
real_filename = os.path.join(real_filename, 'datagrab',
'csvData', '500477.csv')
print(real_filename)
self.assertEqual(csv_filename, real_filename)
def setUp(self):
stock = StockSymbolFactory.create(name='ASHOK Leyland',
symbol='ASHOKLEY',
tickerNumber='500477')
stock.save()
self.assertEquals(StockSymbol.objects.count(), 1)
csv_filename = ut.get_csv_filename(stock)
try:
with transaction.atomic():
count = ut.read_csv_file(stock, csv_filename)
except:
self.fail("exception when trying to insert")
def test_same_data_is_not_inserted_into_db(self):
stock = StockSymbolFactory.create(name='ASHOK Leyland',
symbol='ASHOKLEY',
tickerNumber='500477')
stock.save()
self.assertEquals(StockSymbol.objects.count(), 1)
csv_filename = ut.get_csv_filename(stock)
# first insert
try:
with transaction.atomic():
count = ut.read_csv_file(stock, csv_filename)
except:
self.fail("exception when trying to insert")
self.assertEquals(StockHistory.objects.count(), 1854)
# second insert
try:
with transaction.atomic():
count = ut.read_csv_file(stock, csv_filename)
except:
self.fail("exception when trying to insert")
self.assertEquals(StockHistory.objects.count(), 1854)
def test_derived_csv_filename_is_not_found(self):
stock = StockSymbolFactory.create()
csv_filename = ut.get_csv_filename(stock)
csv_file = ut.read_csv_file(stock, csv_filename)
self.assertEqual(csv_file, False)
