"""Unit tests for utils of Youtube Workflows.""" import datetime from feed_ingestion.util import youtube_util _date = '2018-02-12' def test_get_days(): """Should return current date, start and end dates of reporting. Date range between start and end dates is one week before current date. Current date is not included. """ assert youtube_util.get_days(_date) == ( datetime.date(2018, 2, 12), datetime.date(2018, 2, 12), datetime.date(2018, 2, 18))