def test_more_data_better(): Phi, y, w_gen = generate_fake_data(20000, 20, .5) model_small_data = LinARD(100) model_big_data = LinARD(100) model_small_data.train(Phi[:1000], y[:1000]) model_big_data.train(Phi[:15000], y[:15000]) y_small = model_small_data.predict(Phi[15000:]) y_big = model_big_data.predict(Phi[15000:]) assert np.sum((y[15000:]-y_small)**2) > np.sum((y[15000:]-y_big)**2)