import unittest
import numpy as np
from analysis import Analysis

class testAnalysis(unittest.TestCase):
    """
    A simple test to check the angle means ans error computation.
    """
    def test(self):
        ang = np.array([6,5,6,5,8,4,8,4,7,1,7,1,10,3,10,3,None,None,None,None])
        ang = np.reshape(ang, (5,2,2))
        A = Analysis(ang)
        self.assertEqual(A.results['Mean'][0], 5.5)
        self.assertEqual(A.results['Error'][0], 0.5)