def test_init_weights_shape_is_correct(self):
hmc = HMCNeuron('fake_arg','fake_arg','fake_arg')
weight_count = 3
weights = hmc.init_weights(weight_count)
actual = weights.shape[0]
expected = weight_count
assert(actual == expected)
def test_init_weights_shape_raises_when_too_many_args(self):
hmc = HMCNeuron('fake_arg','fake_arg','fake_arg')
hmc.init_weights(3,2)
def test_init_weights_shape_raises_when_no_args_raises(self):
hmc = HMCNeuron('fake_arg','fake_arg','fake_arg')
hmc.init_weights()