def test_to_matrix(self): ''' Test the to_matrix function with random vectors. ''' for _ in range(ITERATIONS): indim = random_size() vec = np.random.randn(indim) result = to_matrix(vec) self.assertTrue(result.shape == (vec.shape[0], 1)) self.assertTrue((result[:, 0] == vec).any())
def test_add_sample(self): ''' Tests the add_sample method. ''' for _ in range(ITERATIONS): indim = random_size() outdim = random_size() coeff = np.random.randn(indim, outdim) model = UpdatingModel(indim, outdim) for _ in range(SAMPLES): inp = np.random.randn(indim) out = coeff.T.dot(to_matrix(inp))[:, 0] model.add_sample(inp, out) diff = sum(sum(abs(model.coeff.T - coeff))) self.assertTrue(diff < DIFF)