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