KIAPBCI/kiap_bci/tests/test_append.py

import ctypes
import multiprocessing as mp

import numpy as np

import daq2 as daq


def test_writing():
    fname = '/home/ivaz/devel/data/2018-09-06/norm.bin'

    np.random.seed(1)

    ncols = 96
    smp_size = 300
    res0 = np.empty([0,ncols], dtype=np.float32)

    res = np.fromfile(fname, dtype=np.float32)
    res2 = res.reshape(res.size//ncols, ncols)

    for ii in range(res2.shape[0]//smp_size):
        tmp = daq.get_raw()
        res0 = np.append(res0, tmp, axis=0)



    assert np.array_equal(res0, res2), 'np arrays are different'
    return (res0, res2)


test_writing()