def example_time_box(file_name: str):
tb = TimeBox(file_name)
tb._timebox_version = 1
tb._tag_names_are_strings = True
tb._date_differentials_stored = False
tb._num_points = 4
tb._tags = {
'tag_0': TimeBoxTag('tag_0', 1, 'u'),
'tag_1': TimeBoxTag('tag_1', 2, 'i'),
'tag_2_long_name': TimeBoxTag('tag_2_long_name', 4, 'f')
}
tb._start_date = np.datetime64('2018-01-01', 's')
tb._seconds_between_points = 3600
tb._tags['tag_0'].data = np.array([1, 2, 3, 4], dtype=np.uint8)
tb._tags['tag_1'].data = np.array([-4, -2, 0, 2000], dtype=np.int16)
tb._tags['tag_2_long_name'].data = np.array([5.2, 0.8, 3.1415, 8],
dtype=np.float32)
return tb
def test_read_write_file_info_uniform_dates(self):
tb = TimeBox('')
tb._timebox_version = 1
tb._tag_names_are_strings = False
tb._date_differentials_stored = False
tb._num_points = 10
tb._tags = example_tag_definitions()
tb._start_date = np.datetime64('2018-01-01', 's')
tb._seconds_between_points = 3600
file_name = 'test.npb'
with open(file_name, 'wb') as f:
self.assertEqual(231, tb._write_file_info(f))
tb_read = TimeBox('')
with open(file_name, 'rb') as f:
self.assertEqual(231, tb_read._read_file_info(f))
self.assertEqual(tb._timebox_version, tb_read._timebox_version)
self.assertEqual(tb._tag_names_are_strings,
tb_read._tag_names_are_strings)
self.assertEqual(tb._date_differentials_stored,
tb_read._date_differentials_stored)
self.assertEqual(tb._num_points, tb_read._num_points)
self.assertEqual(tb._start_date, tb_read._start_date)
self.assertEqual(tb._seconds_between_points,
tb_read._seconds_between_points)
for t in tb._tags:
self.assertTrue(t in tb_read._tags)
self.assertEqual(tb._tags[t].identifier,
tb_read._tags[t].identifier)
self.assertEqual(tb._tags[t].type_char, tb_read._tags[t].type_char)
self.assertEqual(tb._tags[t].dtype, tb_read._tags[t].dtype)
self.assertEqual(tb._tags[t].bytes_per_value,
tb_read._tags[t].bytes_per_value)
os.remove(file_name)
return
