Changes in / [c00a797a:658dd57] in sasview


Ignore:
File:
1 edited

Legend:

Unmodified
Added
Removed
  • test/sasdataloader/test/utest_abs_reader.py

    raaf5e49 rc551bb3  
    99from sas.sascalc.dataloader.loader import Loader 
    1010from sas.sascalc.dataloader.readers.IgorReader import Reader as IgorReader 
     11from sas.sascalc.dataloader.readers.abs_reader import Reader as AbsReader 
     12from sas.sascalc.dataloader.readers.hfir1d_reader import Reader as HFIRReader 
     13from sas.sascalc.dataloader.readers.danse_reader import Reader as DANSEReader 
     14from sas.sascalc.dataloader.readers.cansas_reader import Reader as CANSASReader 
     15 
    1116from sas.sascalc.dataloader.data_info import Data1D 
    1217  
    1318import os.path 
    1419 
     20 
    1521class abs_reader(unittest.TestCase): 
    1622     
    1723    def setUp(self): 
    18         from sas.sascalc.dataloader.readers.abs_reader import Reader 
    19         self.data = Reader().read("jan08002.ABS") 
    20          
     24        reader = AbsReader() 
     25        self.data = reader.read("jan08002.ABS") 
     26 
    2127    def test_abs_checkdata(self): 
    2228        """ 
     
    6268    def test_checkdata2(self): 
    6369        self.assertEqual(self.data.dy[126], 1) 
    64          
     70 
     71    def test_generic_loader(self): 
     72        # the generic loader should work as well 
     73        data = Loader().load("jan08002.ABS") 
     74        self.assertEqual(data.meta_data['loader'], "IGOR 1D") 
     75 
     76 
    6577class hfir_reader(unittest.TestCase): 
    66      
     78 
    6779    def setUp(self): 
    68         self.data = Loader().load("S2-30dq.d1d") 
    69          
     80        reader = HFIRReader() 
     81        self.data = reader.read("S2-30dq.d1d") 
     82 
    7083    def test_hfir_checkdata(self): 
    7184        """ 
     
    8497        self.assertEqual(self.data.dy[1], 0.000315) 
    8598        self.assertEqual(self.data.dx[1], 0.008249) 
    86          
     99 
     100    def test_generic_loader(self): 
     101        # the generic loader should work as well 
     102        data = Loader().load("S2-30dq.d1d") 
     103        self.assertEqual(data.meta_data['loader'], "HFIR 1D") 
     104 
    87105 
    88106class igor_reader(unittest.TestCase): 
     
    138156     
    139157    def setUp(self): 
    140         self.data = Loader().load("MP_New.sans") 
     158        reader = DANSEReader() 
     159        self.data = reader.read("MP_New.sans") 
    141160 
    142161    def test_checkdata(self): 
     
    172191        self.assertEqual(self.data.err_data[2], 2.06313) 
    173192 
     193    def test_generic_loader(self): 
     194        # the generic loader should work as well 
     195        data = Loader().load("MP_New.sans") 
     196        self.assertEqual(data.meta_data['loader'], "DANSE") 
     197 
    174198  
    175199class cansas_reader(unittest.TestCase): 
    176200     
    177201    def setUp(self): 
     202        reader = CANSASReader() 
     203        data = reader.read("cansas1d.xml") 
     204        self.data = data[0] 
     205 
     206    def test_generic_loader(self): 
     207        # the generic loader should work as well 
    178208        data = Loader().load("cansas1d.xml") 
    179         self.data = data[0] 
    180   
     209        self.assertEqual(data[0].meta_data['loader'], "CanSAS XML 1D") 
     210 
    181211    def test_cansas_checkdata(self): 
    182212        self.assertEqual(self.data.filename, "cansas1d.xml") 
     
    191221            tests won't pass 
    192222        """ 
    193          
    194223        self.assertEqual(self.data.run[0], "1234") 
    195224        self.assertEqual(self.data.meta_data['loader'], "CanSAS XML 1D") 
     
    265294            self.assertEqual(item.distance_unit,'mm') 
    266295 
    267             if item.size.x==50 \ 
    268                 and item.distance==11000.0 \ 
    269                 and item.name=='source' \ 
    270                 and item.type=='radius': 
     296            if item.size.x == 50 \ 
     297                and item.distance == 11000.0 \ 
     298                and item.name == 'source' \ 
     299                and item.type == 'radius': 
    271300                _found1 = True 
    272             elif item.size.x==1.0 \ 
    273                 and item.name=='sample' \ 
    274                 and item.type=='radius': 
     301            elif item.size.x == 1.0 \ 
     302                and item.name == 'sample' \ 
     303                and item.type == 'radius': 
    275304                _found2 = True 
    276305                 
    277         if _found1==False or _found2==False: 
     306        if _found1 == False or _found2 == False: 
    278307            raise RuntimeError, "Could not find all data %s %s" % (_found1, _found2)  
    279308             
     
    312341            print(item.term) 
    313342            for t in item.term: 
    314                 if t['name']=="ABS:DSTAND" \ 
    315                     and t['unit']=='mm' \ 
    316                     and float(t['value'])==1.0: 
     343                if (t['name'] == "ABS:DSTAND" 
     344                    and t['unit'] == 'mm' 
     345                    and float(t['value']) == 1.0): 
    317346                    _found_term2 = True 
    318                 elif t['name']=="radialstep" \ 
    319                     and t['unit']=='mm' \ 
    320                     and float(t['value'])==10.0: 
     347                elif (t['name'] == "radialstep" 
     348                      and t['unit'] == 'mm' 
     349                      and float(t['value']) == 10.0): 
    321350                    _found_term1 = True 
    322351                     
    323         if _found_term1==False or _found_term2==False: 
    324             raise RuntimeError, "Could not find all process terms %s %s" % (_found_term1, _found_term2)  
    325              
    326          
    327          
     352        if _found_term1 == False or _found_term2 == False: 
     353            raise RuntimeError, "Could not find all process terms %s %s" % (_found_term1, _found_term2) 
    328354         
    329355    def test_writer(self): 
    330         from sas.sascalc.dataloader.readers.cansas_reader import Reader 
    331         r = Reader() 
    332         x = np.ones(5) 
    333         y = np.ones(5) 
    334         dy = np.ones(5) 
    335          
     356        r = CANSASReader() 
     357 
    336358        filename = "write_test.xml" 
    337359        r.write(filename, self.data) 
     
    349371        """ 
    350372        filename = "cansas1d_units.xml" 
    351         data = Loader().load(filename) 
     373        data = CANSASReader().read(filename) 
    352374        self.data = data[0] 
    353375        self.assertEqual(self.data.filename, filename) 
     
    360382        """ 
    361383        filename = "cansas1d_badunits.xml" 
    362         data = Loader().load(filename) 
     384        data = CANSASReader().read(filename) 
    363385        self.data = data[0] 
    364386        self.assertEqual(self.data.filename, filename) 
     
    371393        print(self.data.errors) 
    372394        self.assertEqual(len(self.data.errors), 1) 
    373          
    374          
     395 
    375396    def test_slits(self): 
    376397        """ 
     
    378399        """ 
    379400        filename = "cansas1d_slit.xml" 
    380         data = Loader().load(filename) 
     401        data = CANSASReader().read(filename) 
    381402        self.data = data[0] 
    382403        self.assertEqual(self.data.filename, filename) 
     
    400421        self.assertEqual(self.data.run_name['1234'], 'run name') 
    401422        self.assertEqual(self.data.title, "Test title") 
    402          
    403              
     423 
    404424 
    405425if __name__ == '__main__': 
Note: See TracChangeset for help on using the changeset viewer.