Improved test runtime (#11553)
* Remove redundent assert statements and cleanup * Added 'get_date' function * Replace 'freeze_time' with 'mock.patch' * Tox in 185s (py35) * Removed obsolete 'freeze_time' from test_updater * Tox 162s (py35) * Remove test requirement 'freezegun' * Fixed flake8 errors * Added 'mock.patch' for 'feedparser.parse' * Made 'FUNCTION_PATH' a constant * Remove debug statements.
This commit is contained in:
1 parent
f56b3d8e9c
commit
cba55402b1
6 files changed
+66
-70
No files matched your search
@@ -64,7 +64,7 @@ def setup_platform(hass, config, add_devices, discovery_info=None):
|
||||
excludes = config.get(CONF_EXCLUDES)
|
||||
days_offset = config.get(CONF_OFFSET)
|
||||
|
||||
year = (datetime.now() + timedelta(days=days_offset)).year
|
||||
year = (get_date(datetime.today()) + timedelta(days=days_offset)).year
|
||||
obj_holidays = getattr(holidays, country)(years=year)
|
||||
|
||||
if province:
|
||||
@@ -99,6 +99,11 @@ def day_to_string(day):
|
||||
return None
|
||||
|
||||
|
||||
def get_date(date):
|
||||
"""Return date. Needed for testing."""
|
||||
return date
|
||||
|
||||
|
||||
class IsWorkdaySensor(BinarySensorDevice):
|
||||
"""Implementation of a Workday sensor."""
|
||||
|
||||
@@ -156,7 +161,7 @@ class IsWorkdaySensor(BinarySensorDevice):
|
||||
self._state = False
|
||||
|
||||
# Get iso day of the week (1 = Monday, 7 = Sunday)
|
||||
date = datetime.today() + timedelta(days=self._days_offset)
|
||||
date = get_date(datetime.today()) + timedelta(days=self._days_offset)
|
||||
day = date.isoweekday() - 1
|
||||
day_of_week = day_to_string(day)
|
||||
|
||||
|
||||
Reference in new issue
Block a user