def test_GenFile_NotImplementedError(): """Test GenFile throws Error""" from electrolib.sources.genfile import GenFile import py.test # Crete GenFile gf = GenFile([{'label':'chan1'}]) with py.test.raises(NotImplementedError): gf._ft([], []) with py.test.raises(NotImplementedError): gf.get_signals(gf.tstart, gf.tend)
def __init__(self, nchan=10, freq_range=(1, 10), amp=100, *args, **kwargs): def pars_from_f(f): return {'frequency':f, 'label':str(f)[:4] + 'Hz', 'amp':amp} f1, f2 = freq_range fs = numpy.linspace(f1, f2, nchan) fs = [int(f) if f == int(f) else f for f in fs] chan_pars = [pars_from_f(f) for f in fs] GenFile.__init__(self, chan_pars, "Frequency File", *args, **kwargs)
def test_GenFile_close(): """Test GenFile closes without infinite recursion""" from electrolib.sources.genfile import GenFile gf = GenFile([{'label':'chan1'}]) gf.close()