def test_load_google_spreadsheet(self):
"""Test retrieves a Google Spreadsheet. Will fail if no Internet connection."""
# Test without naming a worksheet.
obs = load_google_spreadsheet(self.spreadsheet_key, worksheet_name=None)
self.assertEqual(obs, self.exp_mapping_lines)
# Test with naming a worksheet.
obs = load_google_spreadsheet(self.spreadsheet_key, worksheet_name="Fasting_Map")
self.assertEqual(obs, self.exp_mapping_lines)
def test_load_google_spreadsheet(self):
"""Test retrieves a Google Spreadsheet. Will fail if no Internet connection."""
# Test without naming a worksheet.
obs = load_google_spreadsheet(self.spreadsheet_key,
worksheet_name=None)
self.assertEqual(obs, self.exp_mapping_lines)
# Test with naming a worksheet.
obs = load_google_spreadsheet(self.spreadsheet_key,
worksheet_name='Fasting_Map')
self.assertEqual(obs, self.exp_mapping_lines)
def main():
option_parser, opts, args = parse_command_line_parameters(**script_info)
worksheet_name = opts.worksheet_name
if worksheet_name is not None:
worksheet_name = worksheet_name.strip('"').strip("'")
results = load_google_spreadsheet(opts.spreadsheet_key, worksheet_name)
with open(opts.output_fp, 'w') as out_f:
out_f.write(results)
