def fetch_average(self, current=None):
a_data, b_data = self._fetch_data()
a_data_avg = a_data.mean(axis=0)
b_data_avg = b_data.mean(axis=0)
datasets = [pack_data_set(a_data_avg), pack_data_set(b_data_avg)]
return datasets
def test_pack_unpack_1d(self):
from device_repo.utils import pack_data_set, unpack_data_set
from device_repo import DataSet, DataType
for type in [
np.uint8, np.uint16, np.uint32, np.uint64, np.int8, np.int16,
np.int32, np.int64, np.float, np.double
]:
mock_set = np.array([0, 1, 2, 3, 4], dtype=type)
packed = pack_data_set(mock_set)
assert isinstance(packed, DataSet)
assert packed.shape == (5, )
unpacked = unpack_data_set(packed)
assert isinstance(unpacked, np.ndarray)
assert all(mock_set == unpacked)
def test_pack_unpack_list(self):
from device_repo.utils import pack_data_set, unpack_data_set
from device_repo import DataSet, DataType
mock_set = []
mock_set.append(np.array([0, 1, 2, 3, 4], dtype='byte'))
mock_set.append(np.array([1, 2, 3, 4, 5], dtype='byte'))
mock_set.append(np.array([2, 3, 4, 5, 6], dtype='byte'))
packed = [pack_data_set(mock) for mock in mock_set]
assert all([isinstance(item, DataSet) for item in packed])
assert all([item.shape == (5, ) for item in packed])
assert all([item.type == DataType.Byte for item in packed])
unpacked = unpack_data_set(packed)
assert isinstance(unpacked, list)
for mock, unpacked_item in zip(mock_set, unpacked):
assert all((mock == unpacked_item).flatten())
def fetch(self, current=None):
a_data, b_data = self._fetch_data()
datasets = [pack_data_set(a_data), pack_data_set(b_data)]
return datasets