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)