def test_train_rbm_wrong_shape(self):
r = RBM(5, 10)
data = [[0,0,0,0]]
with self.assertRaises(ValueError):
r.train(data,1,1)
def test_train_rbm_negative_batches(self):
r = RBM(5, 10)
data = [[1,1,1,1,1],[0,0,0,0,0]]
with self.assertRaises(ValueError):
r.train(data,1,-1)
def train_rbm(self):
r = RBM(5, 10)
data = [[1,1,1,1,1],[0,0,0,0,0]]
r.train(data,1,1)
