def test_update_model():
bayes = test_NaiveBayes()
print bayes.model
arr = get_nb_data()
print arr
truth_rows, data_rows, data_mus, y_mu = hw3_run.get_data_and_mus(arr)
bayes.train(data_rows, truth_rows)
print bayes.model
def test_get_data_and_mus():
arr = get_test_data(4)
truth_rows, data_rows, data_mus, y_mu = hw3_run.get_data_and_mus(arr)
print 'unsplit {}'.format(arr)
print 'truth rows {}'.format(truth_rows)
print 'data rows {}'.format(data_rows)
print 'data mus {}'.format(data_mus)
print 'y_mu {}'.format(y_mu)
def test_update_model_change():
bayes = test_NaiveBayes()
print bayes.model
arr = get_nb_data(5)
print arr
truth_rows, data_rows, data_mus, y_mu = hw3_run.get_data_and_mus(arr)
bayes.train(data_rows, truth_rows)
print bayes.model
test = get_nb_test_data(5)
print 'prediction'
print bayes.predict(test)
