[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 |
---|
| 11 | from sas.sascalc.dataloader.file_reader_base_class import FileReader |
---|
| 12 | |
---|
| 13 | logger = logging.getLogger(__name__) |
---|
| 14 | |
---|
| 15 | |
---|
[f53d684] | 16 | def find(filename): |
---|
| 17 | return os.path.join(os.path.dirname(__file__), filename) |
---|
| 18 | |
---|
| 19 | |
---|
[beba407] | 20 | class GenericFileReaderTests(unittest.TestCase): |
---|
| 21 | |
---|
| 22 | def setUp(self): |
---|
[3648cbf] | 23 | self.reader = TestFileReader() |
---|
[f53d684] | 24 | self.bad_file = find("ACB123.txt") |
---|
| 25 | self.good_file = find("123ABC.txt") |
---|
[beba407] | 26 | |
---|
| 27 | def test_bad_file_path(self): |
---|
| 28 | output = self.reader.read(self.bad_file) |
---|
[3648cbf] | 29 | self.assertEqual(output, []) |
---|
[beba407] | 30 | |
---|
| 31 | def test_good_file_path(self): |
---|
[3648cbf] | 32 | f = open(self.good_file, 'w') |
---|
| 33 | f.write('123ABC exists!') |
---|
| 34 | f.close() |
---|
[beba407] | 35 | output = self.reader.read(self.good_file) |
---|
[3648cbf] | 36 | self.assertEqual(len(output), 1) |
---|
| 37 | self.assertEqual(output[0].meta_data["blah"], '123ABC exists!') |
---|
[beba407] | 38 | |
---|
| 39 | def tearDown(self): |
---|
| 40 | if os.path.isfile(self.bad_file): |
---|
| 41 | os.remove(self.bad_file) |
---|
| 42 | if os.path.isfile(self.good_file): |
---|
| 43 | os.remove(self.good_file) |
---|
[3648cbf] | 44 | |
---|
| 45 | class TestFileReader(FileReader): |
---|
| 46 | def get_file_contents(self): |
---|
| 47 | """ |
---|
| 48 | Reader specific class to access the contents of the file |
---|
| 49 | All reader classes that inherit from FileReader must implement |
---|
| 50 | """ |
---|
| 51 | x = np.zeros(0) |
---|
| 52 | y = np.zeros(0) |
---|
| 53 | self.current_dataset = plottable_1D(x,y) |
---|
| 54 | self.current_datainfo = DataInfo() |
---|
| 55 | self.current_datainfo.meta_data["blah"] = self.nextline() |
---|
[f53d684] | 56 | self.send_to_output() |
---|