[beba407] | 1 | """ |
---|
| 2 | Unit tests for the generic file reader class |
---|
| 3 | """ |
---|
| 4 | |
---|
| 5 | import os |
---|
| 6 | import unittest |
---|
| 7 | import logging |
---|
| 8 | import numpy as np |
---|
| 9 | |
---|
| 10 | from sas.sascalc.dataloader.data_info import DataInfo, plottable_1D |
---|
[3ee0451] | 11 | from sas.sascalc.dataloader.loader import Loader |
---|
[425feff] | 12 | from sas.sascalc.dataloader.loader_exceptions import NoKnownLoaderException |
---|
[beba407] | 13 | from sas.sascalc.dataloader.file_reader_base_class import FileReader |
---|
| 14 | |
---|
| 15 | logger = logging.getLogger(__name__) |
---|
| 16 | |
---|
| 17 | |
---|
[f53d684] | 18 | def find(filename): |
---|
| 19 | return os.path.join(os.path.dirname(__file__), filename) |
---|
| 20 | |
---|
| 21 | |
---|
[beba407] | 22 | class GenericFileReaderTests(unittest.TestCase): |
---|
| 23 | |
---|
| 24 | def setUp(self): |
---|
[3648cbf] | 25 | self.reader = TestFileReader() |
---|
[f53d684] | 26 | self.bad_file = find("ACB123.txt") |
---|
| 27 | self.good_file = find("123ABC.txt") |
---|
[3ee0451] | 28 | self.generic_reader = Loader() |
---|
| 29 | self.deprecated_file_type = find("FEB18012.ASC") |
---|
[beba407] | 30 | |
---|
| 31 | def test_bad_file_path(self): |
---|
[425feff] | 32 | self.assertRaises(NoKnownLoaderException, self.reader.read, |
---|
| 33 | self.bad_file) |
---|
[beba407] | 34 | |
---|
| 35 | def test_good_file_path(self): |
---|
[3648cbf] | 36 | f = open(self.good_file, 'w') |
---|
| 37 | f.write('123ABC exists!') |
---|
| 38 | f.close() |
---|
[beba407] | 39 | output = self.reader.read(self.good_file) |
---|
[3648cbf] | 40 | self.assertEqual(len(output), 1) |
---|
| 41 | self.assertEqual(output[0].meta_data["blah"], '123ABC exists!') |
---|
[beba407] | 42 | |
---|
[3ee0451] | 43 | def test_old_file_types(self): |
---|
| 44 | f = self.generic_reader.load(self.deprecated_file_type) |
---|
| 45 | last_f = f[0] |
---|
| 46 | if hasattr(last_f, "errors"): |
---|
| 47 | self.assertEquals(len(last_f.errors), 1) |
---|
| 48 | else: |
---|
| 49 | self.fail("Errors did not propogate to the file properly.") |
---|
| 50 | |
---|
[beba407] | 51 | def tearDown(self): |
---|
| 52 | if os.path.isfile(self.bad_file): |
---|
| 53 | os.remove(self.bad_file) |
---|
| 54 | if os.path.isfile(self.good_file): |
---|
| 55 | os.remove(self.good_file) |
---|
[3648cbf] | 56 | |
---|
| 57 | class TestFileReader(FileReader): |
---|
| 58 | def get_file_contents(self): |
---|
| 59 | """ |
---|
| 60 | Reader specific class to access the contents of the file |
---|
| 61 | All reader classes that inherit from FileReader must implement |
---|
| 62 | """ |
---|
| 63 | x = np.zeros(0) |
---|
| 64 | y = np.zeros(0) |
---|
| 65 | self.current_dataset = plottable_1D(x,y) |
---|
| 66 | self.current_datainfo = DataInfo() |
---|
| 67 | self.current_datainfo.meta_data["blah"] = self.nextline() |
---|
[f53d684] | 68 | self.send_to_output() |
---|