Add unit test for model filtering

qt/python/mantidqt/widgets/samplelogs/test/test_samplelogs_model.py

@@ -7,7 +7,7 @@
 #  This file is part of the mantid workbench.
 #
 #
-from mantid.simpleapi import LoadEventNexus, CreateMDWorkspace
+from mantid.simpleapi import Load, LoadEventNexus, CreateMDWorkspace
 from mantidqt.widgets.samplelogs.model import SampleLogsModel
 
 import unittest
@@ -40,6 +40,9 @@ class SampleLogsModelTest(unittest.TestCase):
         self.assertTrue(model.is_log_plottable("Speed5"))
         self.assertFalse(model.is_log_plottable("duration"))
 
+        self.assertFalse(model.get_is_log_filtered("Speed5"))
+        self.assertFalse(model.get_is_log_filtered("duration"))
+
         stats = model.get_statistics("Speed5")
         self.assertEqual(stats.maximum, 300.0)
 
@@ -56,6 +59,39 @@ class SampleLogsModelTest(unittest.TestCase):
         self.assertEqual(itemModel.item(0,2).text(), "4.0 (2 entries)")
         self.assertEqual(itemModel.item(0,3).text(), "")
 
+    def test_model_with_filtered_data(self):
+        wsTuple = Load('POLREF00014966.nxs')
+        #get the first child workspace of the group
+        ws = wsTuple[0]
+        model = SampleLogsModel(ws)
+
+        log_names = model.get_log_names()
+        self.assertEqual(len(log_names), 152)
+        self.assertIn("raw_uah_log", log_names)
+        self.assertIn("current_period", log_names)
+        self.assertIn("period", log_names)
+
+        values = model.get_log_display_values("raw_uah_log")
+        self.assertEqual(values[0], "raw_uah_log")
+        self.assertEqual(values[1], "float series")
+        self.assertEqual(values[2], "(429 entries)")
+        self.assertEqual(values[3], "uAh")
+
+        self.assertTrue(model.is_log_plottable("raw_uah_log"))
+        self.assertFalse(model.is_log_plottable("current_period"))
+        self.assertTrue(model.is_log_plottable("period"))
+
+        self.assertTrue(model.get_is_log_filtered("raw_uah_log"))
+        self.assertFalse(model.get_is_log_filtered("period"))
+        self.assertFalse(model.get_is_log_filtered("current_period"))
+
+        stats = model.get_statistics("raw_uah_log", filtered = True)
+        self.assertAlmostEqual(stats.maximum, 193.333,3)
+        stats = model.get_statistics("raw_uah_log", filtered = False)
+        self.assertAlmostEqual(stats.maximum, 194.296,3)
+
+        self.assertFalse(model.isMD())
+
     def test_model_MD(self):
         ws1 = LoadEventNexus("CNCS_7860", MetaDataOnly=True)
         ws2 = LoadEventNexus("VIS_19351", MetaDataOnly=True)