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_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 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)