1 | """ |
---|
2 | Unit tests for DataLoader module |
---|
3 | log file "test_log.txt" contains all errors when running loader |
---|
4 | It is create in the folder where test is runned |
---|
5 | """ |
---|
6 | import logging |
---|
7 | logging.basicConfig(level=logging.DEBUG, |
---|
8 | format='%(asctime)s %(levelname)s %(message)s', |
---|
9 | filename='test_log.txt', |
---|
10 | filemode='w') |
---|
11 | |
---|
12 | |
---|
13 | #logger.info('oops I did it again') |
---|
14 | |
---|
15 | import unittest |
---|
16 | import math |
---|
17 | import DataLoader |
---|
18 | from DataLoader.loader import Loader |
---|
19 | from DataLoader.readers import TXT3_Reader,TXT2_Reader |
---|
20 | from DataLoader.readers import IgorReader,danse_reader,tiff_reader |
---|
21 | import os.path |
---|
22 | class testLoader(unittest.TestCase): |
---|
23 | |
---|
24 | |
---|
25 | """ test fitting """ |
---|
26 | #Creating a loader |
---|
27 | L=Loader() |
---|
28 | |
---|
29 | #creating readers |
---|
30 | read1=TXT2_Reader.Reader() |
---|
31 | read2=TXT3_Reader.Reader() |
---|
32 | read3=IgorReader.Reader() |
---|
33 | read4=danse_reader.Reader() |
---|
34 | read5=tiff_reader.Reader() |
---|
35 | #for each readers set an extensions inside the loader |
---|
36 | L.__setitem__('.txt',read2) |
---|
37 | L.__setitem__('.txt',read1) |
---|
38 | L.__setitem__('.dat',read1) |
---|
39 | |
---|
40 | L.__setitem__('.dat',read2) |
---|
41 | L.__setitem__('.ASC',read3) |
---|
42 | L.__setitem__('.sans',read4) |
---|
43 | L.__setitem__('.tif',read5) |
---|
44 | L.__setitem__('.jpg',read5) |
---|
45 | L.__setitem__('.png',read5) |
---|
46 | L.__setitem__('.jpeg',read5) |
---|
47 | L.__setitem__('.gif',read5) |
---|
48 | L.__setitem__('.bmp',read5) |
---|
49 | |
---|
50 | def testLoad0(self): |
---|
51 | """test reading empty file""" |
---|
52 | self.assertEqual(self.L.load('empty.txt'),None) |
---|
53 | |
---|
54 | def testLoad1(self): |
---|
55 | """test reading 2 columns""" |
---|
56 | |
---|
57 | #Testing loading a txt file of 2 columns, the only reader should be read1 |
---|
58 | xload,yload,dyload=self.L.load('test_2_columns.txt') |
---|
59 | x=[2.83954,0.204082,0.408163,0.612245,0.816327,1.02041,1.22449,1.42857,1.63265] |
---|
60 | y=[0.6,3.44938, 5.82026,5.27591,5.2781,5.22531,7.47487,7.85852,10.2278] |
---|
61 | dx=[] |
---|
62 | dy=[] |
---|
63 | self.assertEqual(len(xload),len(x)) |
---|
64 | self.assertEqual(len(yload),len(y)) |
---|
65 | self.assertEqual(len(dyload),0) |
---|
66 | for i in range(len(x)): |
---|
67 | self.assertEqual(xload[i],x[i]) |
---|
68 | self.assertEqual(yload[i],y[i]) |
---|
69 | |
---|
70 | |
---|
71 | def testLoad2(self): |
---|
72 | """Testing loading a txt file of 3 columns""" |
---|
73 | xload,yload,dyload= self.L.load('test_3_columns.txt') |
---|
74 | x=[0,0.204082,0.408163,0.612245,0.816327,1.02041,1.22449] |
---|
75 | y=[2.83954,3.44938,5.82026,5.27591,5.2781,5.22531,7.47487] |
---|
76 | dx=[] |
---|
77 | dy=[0.6,0.676531,0.753061,0.829592,0.906122,0.982653,1.05918] |
---|
78 | self.assertEqual(len(xload),len(x)) |
---|
79 | self.assertEqual(len(yload),len(y)) |
---|
80 | self.assertEqual(len(dyload),len(dy)) |
---|
81 | for i in range(len(x)): |
---|
82 | self.assertEqual(xload[i],x[i]) |
---|
83 | self.assertEqual(yload[i],y[i]) |
---|
84 | self.assertEqual(dyload[i],dy[i]) |
---|
85 | |
---|
86 | |
---|
87 | def testload3(self): |
---|
88 | """ Testing loading Igor data""" |
---|
89 | #tested good file.asc |
---|
90 | Z,xmin, xmax, ymin, ymax= self.L.load('MAR07232_rest.ASC') |
---|
91 | self.assertEqual(xmin,-0.018558945804750416) |
---|
92 | self.assertEqual(xmax, 0.016234058202440633,) |
---|
93 | self.assertEqual(ymin,-0.01684257151702391) |
---|
94 | self.assertEqual(ymax,0.017950440578015116) |
---|
95 | |
---|
96 | #tested corrupted file.asc |
---|
97 | try:self.L.load('AR07232_rest.ASC') |
---|
98 | except ValueError,msg: |
---|
99 | #logging.log(10,str(msg)) |
---|
100 | logging.error(str(msg)) |
---|
101 | def testload4(self): |
---|
102 | """ Testing loading danse file""" |
---|
103 | #tested good file.sans |
---|
104 | data=self.L.load('MP_New.sans') |
---|
105 | |
---|
106 | self.assertEqual(data.__class__,danse_reader.ReaderInfo) |
---|
107 | |
---|
108 | #tested corrupted file.sans |
---|
109 | try: self.L.load('P_New.sans') |
---|
110 | except ValueError,msg: |
---|
111 | #logging.log(40,str(msg)) |
---|
112 | logging.error(str(msg)) |
---|
113 | #else: raise ValueError,"No error raised for missing extension" |
---|
114 | def testload5(self): |
---|
115 | """ Testing loading image file""" |
---|
116 | data=self.L.load('angles_flat.png') |
---|
117 | self.assertEqual(data.__class__,tiff_reader.ReaderInfo) |
---|
118 | |
---|
119 | def testload6(self): |
---|
120 | """test file with unknown extension""" |
---|
121 | try:self.L.load('hello.missing') |
---|
122 | except ValueError,msg: |
---|
123 | self.assertEqual( str(msg),"Unknown file type '.missing'") |
---|
124 | else: raise ValueError,"No error raised for missing extension" |
---|
125 | |
---|
126 | #self.L.lookup('hello.missing') |
---|
127 | try: self.L.lookup('hello.missing') |
---|
128 | except ValueError,msg: |
---|
129 | self.assertEqual( str(msg),"Unknown file type '.missing'") |
---|
130 | else: raise ValueError,"No error raised for missing extension" |
---|
131 | |
---|
132 | def testload7(self): |
---|
133 | """ test file containing an image but as extension .txt""" |
---|
134 | self.assertEqual(self.L.load('angles_flat.txt'),None) |
---|
135 | |
---|