--- a +++ b/python/tests/test_util.py @@ -0,0 +1,29 @@ +# -*- encoding: utf-8 -*- + +from emokit.tasks import EmotivWriterTask +from emokit.util import values_header, writer_task_to_line + +# timestamp + 2 * 14 eeg + 3 gyro = 32 +LINE_LENGTH = 32 + +def _create_EmotivWriterTask(): + data = {} + item = {"value": 1, "quality": 1} + header = set([head.split(" ")[0] for head in values_header.split(",")]) + for head in header: + data[head] = item + timestamp = "2016-12-20 14:10:49.846000" + next_task = EmotivWriterTask(data, timestamp=timestamp) + return next_task + +def test_write_task_to_line(): + next_task = _create_EmotivWriterTask() + line = writer_task_to_line(next_task) + values = line.split(",") + + assert(len(values) == LINE_LENGTH) + +def test_fields(): + header = values_header.split(",") + + assert(len(header) == LINE_LENGTH)