def set_date(self, date_type, date_value):
     assert date_type.lower() in ['start', 'end']
     input_id = 'billing_searchMonthFrom' if date_type.lower() == 'start' else 'billing_searchMonthTo'
     date_field = Find(by=By.ID, value=input_id, context=self)
     date_field.clear()
     date_field.send_keys(date_value)
 def set_date_and_search(self):
     date_field = Find(by=By.CSS_SELECTOR, value='input[ng-model="billingDateFrom"]', context=self)
     date_field.clear()
     date_field.send_keys('2015-01-01')
     self.search_btn_click()
     time.sleep(5)