| 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 IgorReader,danse_reader,tiff_reader |
|---|
| 11 | import os.path |
|---|
| 12 | import os |
|---|
| 13 | class testLoader(unittest.TestCase): |
|---|
| 14 | L=Loader() |
|---|
| 15 | def testplugin(self): |
|---|
| 16 | """ test loading with readers""" |
|---|
| 17 | |
|---|
| 18 | self.assertEqual(self.L.__contains__('.tiff'),True) |
|---|
| 19 | self.assertEqual(self.L.__contains__('.png'),True) |
|---|
| 20 | self.assertEqual(self.L.__contains__('.txt'),True) |
|---|
| 21 | def testplugin1(self): |
|---|
| 22 | """ test loading with plugging""" |
|---|
| 23 | self.L.__setitem__(dir='plugins') |
|---|
| 24 | #Testing loading a txt file of 2 columns, the only reader should be read1 |
|---|
| 25 | xload,yload,dyload=self.L.load('test_2_columns.txt') |
|---|
| 26 | x=[2.83954,0.204082,0.408163,0.612245,0.816327,1.02041,1.22449,1.42857,1.63265] |
|---|
| 27 | y=[0.6,3.44938, 5.82026,5.27591,5.2781,5.22531,7.47487,7.85852,10.2278] |
|---|
| 28 | dx=[] |
|---|
| 29 | dy=[] |
|---|
| 30 | self.assertEqual(len(xload),len(x)) |
|---|
| 31 | self.assertEqual(len(yload),len(y)) |
|---|
| 32 | self.assertEqual(len(dyload),0) |
|---|
| 33 | for i in range(len(x)): |
|---|
| 34 | self.assertEqual(xload[i],x[i]) |
|---|
| 35 | self.assertEqual(yload[i],y[i]) |
|---|
