import unittest
import sys
import cPickle
sys.argv.extend(['1','20','20','70','70','test.pck'])
import agm3
float_type = type(0.1)

class Testagm3(unittest.TestCase):

    def test_load(self):
        p = cPickle.load(open('test.pck'))
        for i in range(100):
            for j in range(100):
                type(p[i][j]) == float_type

suite = unittest.makeSuite(Testagm3)

if __name__ == '__main__':
    unittest.TextTestRunner(verbosity=2).run(suite)