def test_frame_range(self):
sample = dataset.Sequence(test_sequence_fn,
frame_start=10,
frame_stop=20)
self.assertEqual(len(sample), 10)
sample = dataset.Sequence(test_sequence_fn,
frame_start=10,
frame_stop=11)
self.assertEqual(len(sample), 1)
def test_min_max(self):
if dataset.SKIP_FRAMES != 20:
self.skipTest("SKIP_FRAMES must be set to 20")
else:
sample = dataset.Sequence(test_sequence_fn)
self.assertAlmostEqual(sample.temp_min, 23.375)
self.assertAlmostEqual(sample.temp_max, 25.165625)
def test_as_uint8(self):
frame_start = 20
frame_stop = None
sequence_notnormalized = dataset.Sequence(test_fn1,
frame_start=frame_start,
frame_stop=frame_stop,
minmax_normalized=False)
sequence_normalized = dataset.Sequence(test_fn1,
frame_start=frame_start,
frame_stop=frame_stop,
minmax_normalized=True)
self.assertNotEqual(sequence_notnormalized.temp_min, 0)
self.assertNotEqual(sequence_notnormalized.temp_max, 1)
self.assertEqual(sequence_normalized.temp_min, 0)
self.assertEqual(sequence_normalized.temp_max, 1)
self.assertFalse(
np.array_equal(sequence_notnormalized, sequence_normalized))
normalized_uint8 = (255 * sequence_normalized).astype(np.uint8)
self.assertTrue(
np.array_equal(sequence_notnormalized.as_uint8(),
normalized_uint8))
def test_example(self):
if dataset.SKIP_FRAMES != 20:
self.skipTest("SKIP_FRAMES must be set to 20")
frame_start = 20
frame_stop = 23
sequence = dataset.Sequence(test_array_fn,
frame_start=frame_start,
frame_stop=frame_stop)
self.assertEqual(sequence.shape, (3, 16, 16))
self.assertEqual(sequence.temp_min, 0)
self.assertEqual(sequence.temp_max, 2040)
self.assertEqual(sequence.start, frame_start)
self.assertEqual(sequence.stop, frame_stop)
self.assertTrue(np.array_equal(sequence, test_array))
def test_minmax_normalizing(self):
if dataset.SKIP_FRAMES != 20:
self.skipTest("SKIP_FRAMES must be set to 20")
frame_start = 20
frame_stop = 23
sequence = dataset.Sequence(test_array_fn,
frame_start=frame_start,
frame_stop=frame_stop,
minmax_normalized=False)
self.assertTrue(np.array_equal(sequence, test_array))
self.assertEqual(sequence.temp_min, 0)
self.assertEqual(sequence.temp_max, 2040)
sequence = dataset.Sequence(test_array_fn,
frame_start=frame_start,
frame_stop=frame_stop,
minmax_normalized=True)
self.assertTrue(np.array_equal(sequence, test_array_minmax_normalized))
self.assertEqual(sequence.temp_min, 0)
self.assertEqual(sequence.temp_max, 1)
frame_start = 23
frame_stop = 24
sequence = dataset.Sequence(test_array_fn,
frame_start=frame_start,
frame_stop=frame_stop,
minmax_normalized=False)
self.assertTrue(np.array_equal(sequence, 2040 * np.ones([1, 16, 16])))
self.assertEqual(sequence.temp_min, 0)
self.assertEqual(sequence.temp_max, 2040)
sequence = dataset.Sequence(test_array_fn,
frame_start=frame_start,
frame_stop=frame_stop,
minmax_normalized=True)
self.assertTrue(np.array_equal(sequence, np.ones([1, 16, 16])))
self.assertEqual(sequence.temp_min, 0)
self.assertEqual(sequence.temp_max, 1)
def test_frame_start(self):
sample = dataset.Sequence(test_sequence_fn, frame_start=10)
self.assertEqual(len(sample), 388)
def test_attributes(self):
sample = dataset.Sequence(test_sequence_fn)
self.assertEqual(len(sample), 398)
self.assertEqual(sample.filename, test_sequence_fn)
self.assertEqual(sample.sequence_name, "20170203_p1_dark1.csv")