def test_init_momentum_shape_is_correct(self): hmc = HMCNeuron('fake_arg','fake_arg','fake_arg') columns = 3 momentum = hmc.init_momentum(columns) actual = momentum.shape[0] expected = columns assert(actual == expected)
def test_init_momentum_shape_raises_when_too_many_args(self): hmc = HMCNeuron('fake_arg','fake_arg','fake_arg') hmc.init_momentum(3,2)
def test_init_momentum_shape_raises_when_no_args_raises(self): hmc = HMCNeuron('fake_arg','fake_arg','fake_arg') hmc.init_momentum()