Changeset 3a473ef in sasview for test/sasdataloader/test/utest_generic_file_reader_class.py

Timestamp:

Aug 31, 2018 7:13:23 AM (6 years ago)

Author:

Piotr Rozyczko <rozyczko@…>

Branches:

ESS_GUI, ESS_GUI_batch_fitting, ESS_GUI_bumps_abstraction, ESS_GUI_iss1116, ESS_GUI_iss879, ESS_GUI_opencl, ESS_GUI_ordering, ESS_GUI_sync_sascalc

Children:

c1cfa80

Parents:

aa1db44

Message:

Manual update of test/ directory with changes on master - SASVIEW-996

File:

• test/sasdataloader/test/utest_generic_file_reader_class.py (modified) (3 diffs)

test/sasdataloader/test/utest_generic_file_reader_class.py

@@ -8 +8 @@
 import numpy as np
 
-from sas.sascalc.dataloader.data_info import DataInfo, plottable_1D
+from sas.sascalc.dataloader.data_info import DataInfo, plottable_1D, Data1D
+from sas.sascalc.dataloader.loader import Loader
+from sas.sascalc.dataloader.loader_exceptions import NoKnownLoaderException
 from sas.sascalc.dataloader.file_reader_base_class import FileReader
 
@@ -14 +16 @@
 
 
+def find(filename):
+    return os.path.join(os.path.dirname(__file__), filename)
+
+
 class GenericFileReaderTests(unittest.TestCase):
 
     def setUp(self):
-        self.reader = FileReader()
-        self.bad_file = "ACB123.txt"
-        self.good_file = "123ABC.txt"
-        self.msg = "Unable to find file at: {}\n".format(self.bad_file)
-        self.msg += "Please check your file path and try again."
-        x = np.zeros(0)
-        y = np.zeros(0)
-        self.reader.current_dataset = plottable_1D(x, y)
-        self.reader.current_datainfo = DataInfo()
-        self.reader.send_to_output()
+        self.reader = TestFileReader()
+        self.bad_file = find("ACB123.txt")
+        self.good_file = find("123ABC.txt")
+        self.generic_reader = Loader()
+        self.deprecated_file_type = find("FEB18012.ASC")
 
     def test_bad_file_path(self):
-        output = self.reader.read(self.bad_file)
-        self.assertEqual(len(output[0].errors), 1)
-        self.assertEqual(output[0].errors[0], self.msg)
+        self.assertRaises(NoKnownLoaderException, self.reader.read,
+                          self.bad_file)
 
     def test_good_file_path(self):
-        f_open = open(self.good_file, 'w')
-        f_open.close()
+        f = open(self.good_file, 'w')
+        f.write('123ABC exists!')
+        f.close()
         output = self.reader.read(self.good_file)
-        self.assertEqual(len(output[0].errors), 1)
-        self.assertEqual(output[0].errors[0], self.msg)
+        self.assertEqual(len(output), 1)
+        self.assertEqual(output[0].meta_data["blah"], '123ABC exists!')
+
+    def test_old_file_types(self):
+        f = self.generic_reader.load(self.deprecated_file_type)
+        last_f = f[0]
+        if hasattr(last_f, "errors"):
+            self.assertEquals(len(last_f.errors), 1)
+        else:
+            self.fail("Errors did not propogate to the file properly.")
+
+    def test_same_file_unknown_extensions(self):
+        # Five files, all with the same content, but different file extensions
+        no_ext = find("test_data//TestExtensions")
+        not_xml = find("test_data//TestExtensions.notxml")
+        # Deprecated extensions
+        asc_dep = find("test_data//TestExtensions.asc")
+        nxs_dep = find("test_data//TestExtensions.nxs")
+        # Native extension as a baseline
+        xml_native = find("test_data//TestExtensions.xml")
+        # Load the files and check contents
+        no_ext_load = self.generic_reader.load(no_ext)
+        asc_load = self.generic_reader.load(asc_dep)
+        nxs_load = self.generic_reader.load(nxs_dep)
+        not_xml_load = self.generic_reader.load(not_xml)
+        xml_load = self.generic_reader.load(xml_native)
+        self.check_unknown_extension(no_ext_load[0])
+        self.check_unknown_extension(asc_load[0])
+        self.check_unknown_extension(nxs_load[0])
+        self.check_unknown_extension(not_xml_load[0])
+        self.check_unknown_extension(xml_load[0])
+        # Be sure the deprecation warning is passed with the file
+        self.assertEquals(len(asc_load[0].errors), 1)
+        self.assertEquals(len(nxs_load[0].errors), 1)
+
+    def check_unknown_extension(self, data):
+        self.assertTrue(isinstance(data, Data1D))
+        self.assertEquals(len(data.x), 138)
+        self.assertEquals(data.sample.ID, "TK49 c10_SANS")
+        self.assertEquals(data.meta_data["loader"], "CanSAS XML 1D")
 
     def tearDown(self):
@@ -45 +84 @@
         if os.path.isfile(self.good_file):
             os.remove(self.good_file)
+
+class TestFileReader(FileReader):
+    def get_file_contents(self):
+        """
+        Reader specific class to access the contents of the file
+        All reader classes that inherit from FileReader must implement
+        """
+        x = np.zeros(0)
+        y = np.zeros(0)
+        self.current_dataset = plottable_1D(x,y)
+        self.current_datainfo = DataInfo()
+        self.current_datainfo.meta_data["blah"] = self.nextline()
+        self.send_to_output()