def test_filename(self): exFile = ExcelWriter('') self.assertEqual('', exFile.filename) exFile = ExcelWriter('/Users/developer/Downloads/test.xlsx') self.assertEqual('/Users/developer/Downloads/test.xlsx', exFile.filename) self.assertIsNotNone(exFile._workbook) self.assertIsNotNone(exFile._worksheet) self.assertEqual('Sheet', exFile._activeSheetName) self.assertEqual(1, exFile._rowNum) self.assertEqual(1, exFile._column)
def test_set_active_sheet(self): exFile = ExcelWriter('/Users/developer/Downloads/test.xlsx') def test1(): exFile.setActiveSheet('abc') self.assertEqual('abc', exFile.activeSheetName) self.assertIsNotNone(exFile._workbook) self.assertIsNotNone(exFile._worksheet) self.assertIsNotNone(exFile._rowNum) self.assertEqual(1, exFile._rowNum) self.assertIsNotNone(exFile._column) self.assertEqual(1, exFile._column) def test2(): exFile.setActiveSheet('def') self.assertEqual('def', exFile.activeSheetName) self.assertIsNotNone(exFile._workbook) self.assertIsNotNone(exFile._worksheet) self.assertIsNotNone(exFile._rowNum) self.assertEqual(1, exFile._rowNum) self.assertIsNotNone(exFile._column) test1() self.assertTrue('abc' in exFile._workbook.sheetnames) test2() self.assertTrue('abc' in exFile._workbook.sheetnames) self.assertTrue('def' in exFile._workbook.sheetnames)
def test_write_line(self): exFile = ExcelWriter('/Users/developer/Downloads/test.xlsx') def test1(): exFile.setActiveSheet('abc') header = True for item in self._verifyList: exFile.writeLine(item, header) if header: header = False else: header = True self.assertEqual(len(self._verifyList), exFile._worksheet.max_row) self.assertEqual(len(self._verifyList) + 1, exFile._rowNum) self.assertEqual(1, exFile._column) self.assertEqual(3, exFile._worksheet.max_column) for i in range(1, exFile._worksheet.max_row): for j in range(1, 3): currCell = exFile._worksheet.cell(row=i, column=j) self.assertEqual(self._verifyList[i - 1][j - 1], currCell.value) if j <= 2: self.assertEqual('@', currCell.number_format) else: self.assertEqual('0', currCell.number_format) def test2(): exFile.setActiveSheet('def') header = True for item in self._verifyList[:-1]: exFile.writeLine(item, header) if header: header = False else: header = True self.assertEqual( len(self._verifyList) - 1, exFile._worksheet.max_row) self.assertEqual(len(self._verifyList), exFile._rowNum) self.assertEqual(1, exFile._column) self.assertEqual(3, exFile._worksheet.max_column) for i in range(1, exFile._worksheet.max_row): for j in range(1, 3): currCell = exFile._worksheet.cell(row=i, column=j) self.assertEqual(self._verifyList[i - 1][j - 1], currCell.value) test1() self.assertTrue('abc' in exFile._workbook.sheetnames) test2() self.assertTrue('abc' in exFile._workbook.sheetnames) self.assertTrue('def' in exFile._workbook.sheetnames)
def test_save(self): exFile = ExcelWriter('/Users/developer/Downloads/test.xlsx') def test1(): exFile.setActiveSheet('abc') header = True for item in self._verifyList: exFile.writeLine(item, header) if header: header = False else: header = True self.assertEqual(len(self._verifyList), exFile._worksheet.max_row) self.assertEqual(len(self._verifyList) + 1, exFile._rowNum) self.assertEqual(1, exFile._column) self.assertEqual(3, exFile._worksheet.max_column) def test2(): exFile.setActiveSheet('def') header = True for item in self._verifyList[:-1]: exFile.writeLine(item, header) if header: header = False else: header = True self.assertEqual( len(self._verifyList) - 1, exFile._worksheet.max_row) self.assertEqual(len(self._verifyList), exFile._rowNum) self.assertEqual(1, exFile._column) self.assertEqual(3, exFile._worksheet.max_column) test1() test2() self.assertTrue('abc' in exFile._workbook.sheetnames) self.assertTrue('def' in exFile._workbook.sheetnames) exFile.save()
def test3(): val = 'abc' self.assertIsInstance(val, str) ExcelWriter._setFormatting(currCell, val) self.assertEqual('@', currCell.number_format)
def test2(): val = datetime.strptime('1/1/2020', '%m/%d/%Y') self.assertIsInstance(val, datetime) ExcelWriter._setFormatting(currCell, val) self.assertEqual('yyyy-mm-dd', currCell.number_format)
def test1(): val = 1 self.assertIsInstance(val, int) ExcelWriter._setFormatting(currCell, val) self.assertEqual('0', currCell.number_format)
def test_activeSheetName(self): exFile = ExcelWriter('/Users/developer/Downloads/test.xlsx') self.assertEqual('Sheet', exFile.activeSheetName)
def test_openExisting(self): # copyfile('/Users/developer/Downloads/test.xlsx', # '/Users/developer/Downloads/test2.xlsx') exFile = ExcelWriter('/Users/developer/Downloads/test.xlsx', openExisting=True) def test1(): self.assertEqual('/Users/developer/Downloads/test.xlsx', exFile.filename) self.assertIsNotNone(exFile._workbook) self.assertIsNotNone(exFile._worksheet) self.assertEqual('Sheet', exFile._activeSheetName) self.assertEqual(1, exFile._rowNum) self.assertEqual(1, exFile._column) def test2(): # Remove existing sheet 'abc' before proceeding. self.assertTrue('abc' in exFile._workbook.sheetnames) exFile._workbook.remove(exFile._workbook['abc']) self.assertFalse('abc' in exFile._workbook.sheetnames) exFile.setActiveSheet('abc') header = True for item in self._verifyListNew: exFile.writeLine(item, header) if header: header = False else: header = True self.assertEqual(len(self._verifyListNew), exFile._worksheet.max_row) self.assertEqual(len(self._verifyListNew) + 1, exFile._rowNum) self.assertEqual(1, exFile._column) self.assertEqual(3, exFile._worksheet.max_column) def test3(): # Remove existing sheet 'def' before proceeding. self.assertTrue('def' in exFile._workbook.sheetnames) exFile._workbook.remove(exFile._workbook['def']) self.assertFalse('def' in exFile._workbook.sheetnames) exFile.setActiveSheet('def') header = True for item in self._verifyListNew[:-1]: exFile.writeLine(item, header) if header: header = False else: header = True self.assertEqual( len(self._verifyListNew) - 1, exFile._worksheet.max_row) self.assertEqual(len(self._verifyListNew), exFile._rowNum) self.assertEqual(1, exFile._column) self.assertEqual(3, exFile._worksheet.max_column) test1() test2() test3() self.assertTrue('Sheet' in exFile._workbook.sheetnames) self.assertTrue('abc' in exFile._workbook.sheetnames) self.assertTrue('def' in exFile._workbook.sheetnames) exFile.save('/Users/developer/Downloads/test2.xlsx')
python3 controller/Covid19.py --download --location "USA/New York/New York, USA/New York/Nassau, USA/New York/Westchester, USA/New York/Rockland, USA/New Jersey/Bergen, USA/New Jersey/Hudson, USA/New Jersey/Middlesex, USA/New Jersey/Mercer, USA/Connecticut/Fairfield, USA/Washington/King, USA/California/Santa Clara, USA/Illinois/Cook, United Kingdom//, India//" --templateOutput "/Users/developer/Downloads/covid19Template7.xlsm" --output "/Users/developer/Downloads/covid19/covid19.xlsm" """ args = Covid19Args() args.setup() args.parse() inputList = Covid19ControllerUtils.setupInputsFromArgs(args) obsColl = Covid19ControllerUtils.createObsCollection(inputList) writer = ExcelWriter(args.templateOutput, openExisting=True, hasMacros=True) writer._rowNum = 2 print('xx', str(args.locationList)) for locStr in args.locationList: # print('xx', locStr) loc = CmdLineToLocation.makeLocation(locStr) print('Adding values for location "{0}".'.format(loc.locationAsStr())) for obs in obsColl.getObservations(loc): # outputTup = (loc.country, loc.state, loc.county, '', obs.obsDate.value.strftime('%m/%d/%Y'), obs.value) outputTup = (loc.country, loc.state, loc.county, '', loc.locationAsStr(), obs.obsDate.value, obs.value, obs.log2Value, obs.lnValue) writer.writeLine(outputTup, header_=False) writer.save(args.outputfile)