def test_test(mocker):
pd = PerceptualDiscrimination(dt = 10, tau = 100, T = 2000, N_batch = 128)
gen = pd.batch_generator()
x,y,m,p = next(gen)
params = get_params()
rnn = RNN(params)
mocker.patch.object(RNN, 'forward_pass')
RNN.forward_pass.return_value = tf.fill([params['N_batch'], params['N_steps'], params['N_out']], float('nan')), tf.fill([params['N_batch'], params['N_steps'], params['N_rec']], float('nan'))
rnn.test(x)
def test_test(mocker):
rdm = rd.RDM(dt=10, tau=100, T=2000, N_batch=128)
gen = rdm.batch_generator()
x, y, m, p = next(gen)
params = get_params()
rnn = RNN(params)
with pytest.raises(UserWarning) as excinfo:
rnn.test(x)
assert 'build' in str(excinfo.value)
mocker.patch.object(RNN, 'forward_pass')
RNN.forward_pass.return_value = tf.fill(
[params['N_batch'], params['N_steps'], params['N_out']],
float('nan')), tf.fill(
[params['N_batch'], params['N_steps'], params['N_rec']],
float('nan'))
rnn.build()
rnn.test(x)
