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()
