from data_loaders import load_data
import numpy as np
from scipy import stats
from matplotlib import pyplot as plt
from astroML.plotting import setup_text_plots
from astroML.stats import mean_sigma, median_sigmaG

setup_text_plots(fontsize=8, usetex=False)

if __name__ == '__main__':
    quasars = load_data.get_data('../data/I_Paris_plate_mjd_fiber.dat')
    qsos, wavelengths = load_data.build_rest_frame(quasars)
    qsos_m = load_data.list_to_matrix(qsos)

    for i in xrange(0, 400, 50):
        fig = plt.figure(figsize=(5, 7))
        fig.subplots_adjust(left=0.13, right=0.95,
                            bottom=0.06, top=0.95,
                            hspace=0.1)
        ax = fig.add_subplot(1, 1, 1)

        avg = np.mean(qsos_m[:, i])
        std = np.std(qsos_m[:, i])
        data = (qsos_m[:, i] - avg) / std

        x = np.linspace(-5, 5, 1000)
        pdf = stats.norm(0, 1).pdf(x)

        A2, sig, crit = stats.anderson(data)
        D, pD = stats.kstest(data, "norm")
        W, pW = stats.shapiro(data)
from data_loaders import load_data

if __name__ == '__main__':
    qso_dict = load_data.get_data('../data/I_Paris_plate_mjd_fiber.dat')