def printWithWellProdDate(self,wellID,prodDate,product): db = DataBase(config.get_file_dir() + 'database.xlsx') md = db.getMonthlyDataByWellProdMonthProduct(wellID,prodDate,product) well = db.getWell(wellID) royalty = db.getRoyaltyMaster(well.lease) lease = db.getLease(well.lease) calc = db.getCalcDataByWellProdMonthProduct(wellID,prodDate,product) ws = RoyaltyWorksheet() ws.printSaskOilRoyaltyRate(md,well,royalty,lease,calc)
class TestLoader(unittest.TestCase): TEST_SPREADSHEET = config.get_file_dir() + 'test_database.xlsx' def test_run(self): #Testing loading an Excel spreadsheet into an sqlite3 database. # print ("Creating temporary database %s" % self.TEST_DATABASE) self.assertEqual(config.get_environment(),'unittest') # Distructive Tests must run in unittest enviornment dbu = DatabaseUtilities() dbc = DatabaseCreate() loader = Loader() dbu.delete_all_tables() dbc.create_all() dbi = config.get_database_instance() loader.connect() loader.open_excel(self.TEST_SPREADSHEET) shower = Shower() shower.connect() #Test that the worksheet has x number of tabs self.assertEqual(len(loader.wb.get_sheet_names()), 2) #Test that each tab has x number of columns self.assertEqual(len(loader.wb['Well'].columns), 12) self.assertEqual(len(loader.wb['Royalty Master'].columns), 11) #Test that each tab has x number of rows self.assertEqual(len(loader.wb['Well'].rows), 9) self.assertEqual(len(loader.wb['Royalty Master'].rows), 11) # print(dbi.get_table_names()) # self.assertEqual(len(dbi.get_table_names()), 0) # # #Test that we have x number of tables loader.load_all_sheets() # self.assertEqual(len(dbi.get_table_names()), 2) #test that each table has x number of columns self.assertEqual(len(shower.show_table('Well')), 8) self.assertEqual(len(shower.show_table('RoyaltyMaster')), 10) #test that each table has x number of row self.assertEqual(len(shower.show_columns('Well')), 12) self.assertEqual(len(shower.show_columns('RoyaltyMaster')), 11) #test column type self.assertEqual(shower.column_type('Well', 'ID'), 'integer') loader.commit()
class TestLoader(unittest.TestCase): TEST_SPREADSHEET = config.get_file_dir() + 'test_database.xlsx' # # Test that the load works properly def test_run(self): self.assertEqual( config.get_environment(), 'unittest') # Distructive Tests must run in unittest enviornment dbu = DatabaseUtilities() # dbc = DatabaseCreate() loader = Loader() dbu.delete_all_tables() dbi = config.get_database_instance() db = config.get_database() loader.open_excel(self.TEST_SPREADSHEET) # Test that the worksheet has x number of tabs self.assertEqual(len(loader.wb.get_sheet_names()), 2) # Test that each tab has x number of columns self.assertEqual(len(loader.wb['Well'].columns), 12) self.assertEqual(len(loader.wb['Royalty Master'].columns), 11) # Test that each tab has x number of rows self.assertEqual(len(loader.wb['Well'].rows), 9) self.assertEqual(len(loader.wb['Royalty Master'].rows), 11) loader.load_all_sheets() # Test that we have x number of tables self.assertEqual(len(dbi.get_table_names()), 2) # check the rows and columns for well rows = db.select('Well') self.assertEqual(len(rows), 8) self.assertEqual(len(dbi.get_column_names()), 12) # check the rows and columns for royalty master rows = db.select('RoyaltyMaster') self.assertEqual(len(rows), 10) self.assertEqual(len(dbi.get_column_names()), 11)
def test_get_file_dir(self): self.assertTrue(os.path.isdir(config.get_file_dir()), 'File directory must exist: ' + config.get_file_dir())
def test_getMonthlyDataByWellProdMonthProduct(self): db = DataBase(config.get_file_dir() + 'database new.xlsx') md = db.getMonthlyDataByWellProdMonthProduct(6, 201501, 'Oil') self.assertRaises(AppError, db.getMonthlyDataByWellProdMonthProduct, 99999, 201512, 'Oil')
elif cell.value is None: data = data + " Null," else: raise AppError('*** Not Loaded',cell.value, type(cell.value)) i += 1 data = data + ')' data = data.replace(',)', ')') insert = insert + data print(insert) self.dbi.execute(insert) def commit (self): self.dbi.commit() def close(self): self.dbi.close() if __name__ == '__main__': None worksheet = config.get_file_dir() + 'database.xlsx' config.set_database_name('browse.db') loader = Loader() loader.open_excel(worksheet) loader.load_all_sheets() loader.close()
def test_get_file_dir(self): self.assertTrue(os.path.isdir(config.get_file_dir()),'File directory must exist: ' + config.get_file_dir())
def test_getMonthlyDataByWellProdMonthProduct(self): db = DataBase(config.get_file_dir() + 'database new.xlsx') md = db.getMonthlyDataByWellProdMonthProduct(6,201501,'Oil') self.assertRaises(AppError,db.getMonthlyDataByWellProdMonthProduct,99999,201512,'Oil')
# data = data + str(cell.value.strftime("%Y/%m/%d")) + "," elif cell.value is None: data = data + " Null," else: raise AppError('*** Not Loaded', cell.value, type(cell.value)) i += 1 data = data + ')' data = data.replace(',)', ')') insert = insert + data print(insert) self.dbi.execute(insert) def commit(self): self.dbi.commit() def close(self): self.dbi.close() if __name__ == '__main__': None worksheet = config.get_file_dir() + 'database.xlsx' config.set_database_name('browse.db') loader = Loader() loader.open_excel(worksheet) loader.load_all_sheets() loader.close()