def test_get_image_good_datetime(self):
"""Test ts_get_image with a datetime obj on a good timestream"""
for iii in range(len(helpers.TS_MANIFOLD_DATES)):
date = ts_parse_date(helpers.TS_MANIFOLD_DATES[iii])
ts = helpers.FILES["timestream_manifold"]
res = ts_get_image(ts, date)
self.assertEqual(res, helpers.TS_MANIFOLD_FILES_JPG[iii])
def test_get_image_good_str(self):
"""Test ts_get_image with a str date on a good timestream"""
for iii in range(len(helpers.TS_DATES)):
date = helpers.TS_DATES[iii]
ts = helpers.FILES["timestream"]
res = ts_get_image(ts, date)
self.assertEqual(res, helpers.TS_FILES_JPG[iii])
def test_get_image_bad_params(self):
"""Test giving bad paramters to ts_get_image raises ValueError"""
with self.assertRaises(ValueError):
# bad ts_path param
ts_get_image(None, helpers.TS_MANIFOLD_DATES[0])
with self.assertRaises(ValueError):
# bad date param
ts_get_image(helpers.FILES["timestream_manifold"], None)
with self.assertRaises(ValueError):
# unparseable str date param
ts_get_image(helpers.FILES["timestream_manifold"], "NOTADATE")
with self.assertRaises(ValueError):
# bad subsecond param
ts_get_image(helpers.FILES["timestream_manifold"],
helpers.TS_MANIFOLD_DATES[0], n="this should be an int")
def test_get_image_missing_datetime(self):
"""Test ts_get_image with a missing datetime on a good timestream"""
date = ts_parse_date("2010_10_10_10_10_10")
ts = helpers.FILES["timestream_manifold"]
res = ts_get_image(ts, date)
self.assertEqual(res, None)
def test_get_image_missing_str(self):
"""Test ts_get_image with a missing str date on a good timestream"""
date = "2010_10_10_10_10_10"
ts = helpers.FILES["timestream_manifold"]
res = ts_get_image(ts, date)
self.assertEqual(res, None)
