''' Suggested solution for Exercise 18.6'''

# from __future__ import print_function
import numpy as np
import matplotlib.pyplot as plt
from python18_examples import breweries
from python18_1 import fct

x, y1, y2, y3, y4 = breweries()

[X, beta1, yhat, ybar, Syy, SSE, SSR, R21] =  fct(x,y1)  # BBAG
[X, beta2, yhat, ybar, Syy, SSE, SSR, R22] =  fct(x,y2)  # Heineken
[X, beta3, yhat, ybar, Syy, SSE, SSR, R23] =  fct(x,y3)  # Carlsberg
[X, beta4, yhat, ybar, Syy, SSE, SSR, R24] =  fct(x,y4)  # Holsten

t = np.arange(0, 14, 1e-2)

plt.figure(1)
plt.scatter(x,y1)
plt.plot(t, beta1[0] + beta1[1]*t, color='orange')
plt.axis([0, 14, 80, 160])
plt.title('BBAG')

plt.figure(2)
plt.scatter(x,y2)
plt.plot(t, beta2[0] + beta2[1]*t, color='orange')
plt.axis([0, 14, 80, 160])
plt.title('Heineken')

plt.figure(3)
plt.scatter(x,y3)
plt.plot(t, beta3[0] + beta3[1]*t, color='orange')
plt.axis([0, 14, 80, 160])
plt.title('Carlsberg')

plt.figure(4)
plt.scatter(x,y4)
plt.plot(t, beta4[0] + beta4[1]*t, color='orange')
plt.axis([0, 14, 80, 160])
plt.title('Holsten')

print('\n')
print('R^2(BBAG) = ', R21)
print('R^2(Heineken) = ', R22)
print('R^2(Carlsberg) = ', R23)
print('R^2(Holsten) = ', R24)
print('\nThere is no evidence that Carlsberg has a more significant trend than'
      + ' BBAG.\n')

plt.show()

#raw_input()