[aa749ac] | 1 | """ |
---|
| 2 | Unit tests for DataLoader module |
---|
| 3 | """ |
---|
| 4 | |
---|
| 5 | import unittest |
---|
| 6 | import math |
---|
| 7 | import DataLoader |
---|
| 8 | from DataLoader.loader import Loader |
---|
| 9 | from DataLoader.readers import TXT3_Reader,TXT2_Reader |
---|
| 10 | from DataLoader.readers import DataReader,danse_reader,tiff_reader |
---|
| 11 | |
---|
| 12 | class testLoader(unittest.TestCase): |
---|
| 13 | """ test fitting """ |
---|
| 14 | def testLoad(self): |
---|
| 15 | """ |
---|
| 16 | test module Load |
---|
| 17 | """ |
---|
| 18 | #Creating a loader |
---|
| 19 | L=Loader() |
---|
| 20 | |
---|
| 21 | #creating readers |
---|
| 22 | read1=TXT2_Reader.Reader() |
---|
| 23 | read2=TXT3_Reader.Reader() |
---|
| 24 | read3=DataReader.Reader() |
---|
| 25 | read4=danse_reader.Reader() |
---|
| 26 | read5=tiff_reader.Reader() |
---|
| 27 | #for each readers set an extensions inside the loader |
---|
| 28 | L.__setitem__('.txt',read1) |
---|
| 29 | L.__setitem__('.dat',read1) |
---|
| 30 | L.__setitem__('.txt',read2) |
---|
| 31 | L.__setitem__('.dat',read2) |
---|
| 32 | L.__setitem__('.ASC',read3) |
---|
| 33 | L.__setitem__('.sans',read4) |
---|
| 34 | L.__setitem__('.tif',read5) |
---|
| 35 | L.__setitem__('.jpg',read5) |
---|
| 36 | L.__setitem__('.png',read5) |
---|
| 37 | L.__setitem__('.jpeg',read5) |
---|
| 38 | L.__setitem__('.gif',read5) |
---|
| 39 | L.__setitem__('.bmp',read5) |
---|
| 40 | |
---|
| 41 | #test reading empty file |
---|
| 42 | self.assertEqual(L.load('empty.txt'),None) |
---|
| 43 | |
---|
| 44 | #print L.loadData('test_2_columns.txt') |
---|
| 45 | xload,yload,dyload=L.load('test_2_columns.txt') |
---|
| 46 | x=[.83954,0.204082,0.408163,0.612245,0.816327,1.02041,1.22449,1.42857,1.63265] |
---|
| 47 | y=[0.6,3.44938, 5.82026,5.27591,5.2781,5.22531,7.47487,7.85852,10.2278] |
---|
| 48 | dx=[] |
---|
| 49 | dy=[] |
---|
| 50 | self.assertEqual(len(xload),len(x)) |
---|
| 51 | self.assertEqual(len(yload),len(y)) |
---|
| 52 | self.assertEqual( L.getAcTReader('test_2_columns.txt'),DataLoader.readers.TXT2_Reader.Reader) |
---|
| 53 | |
---|
| 54 | |
---|
| 55 | |
---|
| 56 | |
---|
| 57 | |
---|