def test_ImageStimulus_encode():
stim = ImageStimulus(np.random.rand(4, 5))
# Amplitude encoding in default range:
enc = stim.encode()
npt.assert_almost_equal(enc.time[-1], 500)
npt.assert_almost_equal(enc.data.max(axis=1).min(), 0)
npt.assert_almost_equal(enc.data.max(axis=1).max(), 50)
# Amplitude encoding in custom range:
enc = stim.encode(amp_range=(2, 43))
npt.assert_almost_equal(enc.time[-1], 500)
npt.assert_almost_equal(enc.data.max(axis=1).min(), 2)
npt.assert_almost_equal(enc.data.max(axis=1).max(), 43)
with pytest.raises(TypeError):
stim.encode(pulse={'invalid': 1})
with pytest.raises(ValueError):
stim.encode(pulse=LogoUCSB())
def test_LogoUCSB():
logo = LogoUCSB()
npt.assert_equal(logo.img_shape, (324, 727))
npt.assert_equal(logo.time, None)
npt.assert_almost_equal(logo.data.min(), 0)
npt.assert_almost_equal(logo.data.max(), 1)