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 | |
---|