def test_list_validation():
dv = DataValidation(ValidationType.LIST, formula1='"Dog,Cat,Fish"')
assert dv.formula1, '"Dog,Cat == Fish"'
assert dv.generate_attributes_map()['type'] == 'list'
assert dv.generate_attributes_map()['allowBlank'] == '0'
assert dv.generate_attributes_map()['showErrorMessage'] == '1'
assert dv.generate_attributes_map()['showInputMessage'] == '1'
def test_data_validation(out, doc, worksheet):
from ..worksheet import write_worksheet_datavalidations
from openpyxl.datavalidation import DataValidation, ValidationType
ws = worksheet
dv = DataValidation(ValidationType.LIST, formula1='"Dog,Cat,Fish"')
dv.add_cell(ws['A1'])
ws.add_data_validation(dv)
write_worksheet_datavalidations(doc, worksheet)
xml = out.getvalue()
expected = """
<dataValidations count="1">
<dataValidation allowBlank="0" showErrorMessage="1" showInputMessage="1" sqref="A1" type="list">
<formula1>"Dog,Cat,Fish"</formula1>
<formula2>None</formula2>
</dataValidation>
</dataValidations>
"""
diff = compare_xml(xml, expected)
assert diff is None, diff
scenarioName = row.sName
print 'Creating the EMFAC 2014 User Defined Input Workbook and Worksheets'
book = Workbook()
sheet_settings = book.active
sheet_settings.title = 'Settings'
sheet_scen_vmt = book.create_sheet()
sheet_scen_vmt.title = ('Daily_VMT_By_Veh_Tech')
sheet_scen_speed = book.create_sheet()
sheet_scen_speed.title = 'Hourly_Fraction_Veh_Tech_Speed'
####WRITE REGIONAL SCENARIOS SHEETS####
#Write Column Headers for Regional Scenarios Sheet
print 'Writing header row for Settings worksheet'
dv = DataValidation(validation_type="list", formula1='"Annual,Summer,Winter,January,February,March,April,May,June,July,August,September,October,November,December"', allow_blank=True)
sheet_settings.add_data_validation(dv)
sheet_settings.cell(row = 0, column = 0).value = 'Parameter'
sheet_settings.cell(row = 0, column = 1).value = 'Value'
sheet_settings.cell(row = 1, column = 0).value = 'Date'
sheet_settings.cell(row = 1, column = 1).value = datetime.now().strftime('%x %X')
sheet_settings.cell(row = 2, column = 0).value = 'Season/Month'
sheet_settings.cell(row = 2, column = 1).value = season
dv.add_cell(sheet_settings.cell(row = 2, column = 1))
sheet_settings.cell(row = 3, column = 0).value = 'SB375 Run'
sheet_settings.cell(row = 3, column = 1).value = sb375
def test_prompt_message():
dv = DataValidation(ValidationType.LIST, formula1='"Dog,Cat,Fish"')
dv.set_prompt_message('Please enter a value')
assert dv.generate_attributes_map()['promptTitle'] == 'Validation Prompt'
assert dv.generate_attributes_map()['prompt'] == 'Please enter a value'
def test_error_message():
dv = DataValidation(ValidationType.LIST, formula1='"Dog,Cat,Fish"')
dv.set_error_message('You done bad')
assert dv.generate_attributes_map()['errorTitle'] == 'Validation Error'
assert dv.generate_attributes_map()['error'] == 'You done bad'
