source: sasview/test/projectsaveandload/test/utest_project.py @ 08f921e

"""
    Unit tests for saving and loading .svs files
"""

import os
import sys
import unittest
import warnings
import threading

from sas.sascalc.dataloader.loader import Loader
from sas.sascalc.fit.pagestate import Reader as fit_state_reader
from sas.sasgui.perspectives.invariant.invariant_state import Reader as invariant_reader
from sas.sasgui.perspectives.pr.inversion_state import Reader as pr_reader
from sas.sasgui.perspectives.corfunc.corfunc_state import Reader as corfunc_reader

warnings.simplefilter("ignore")

TEMP_FOLDER = "temp_folder"
STATE_LOADERS = [fit_state_reader, invariant_reader, pr_reader, corfunc_reader]


class projects(unittest.TestCase):

    def setUp(self):
        """
        Set up the base unit test class and variables used throughout the tests
        """
        self.addCleanup(self.remove_dir)
        self.loader = Loader()
        self.data1d = self.loader.load("test_data/data1D.h5")
        self.data2d = self.loader.load("test_data/data2D.h5")
        self.sasviewThread = sasviewThread()
        self.sasviewThread.start_local()
        if not (os.path.isdir(TEMP_FOLDER)):
            os.makedirs(TEMP_FOLDER)

    def tearDown(self):
        self.remove_dir()
        if hasattr(self.sasviewThread, "isAlive"):
            if self.sasviewThread.isAlive():
                print("TODO: Close app directly")
                self.app.gui.Close()
                pass

    def remove_dir(self):
        if(os.path.isdir(TEMP_FOLDER)):
            os.removedirs(TEMP_FOLDER)

    def test_saveload_data1d_fitting_only(self):
        """
        Test saving and loading a project with a single data set sent to fitting
        """
        self.sasviewThread.join(5)
        self.assertTrue(1 == 1)

        # TODO: Send 1D to fitting, select model, save project, load project
        # TODO: Send both to fitting, select model on both, save/load
        # TODO: Send both to fitting, select model on one, save/load
        # TODO: Send 1D to each other persepective, save/load
        # TODO: Send 1D to every perspective, save/load
        # TODO: Save/load simultaneous/constrained fit project


class sasviewThread(threading.Thread):
    """Run the MainLoop as a thread. Access the frame with self.frame."""

    def __init__(self, autoStart=True):
        threading.Thread.__init__(self)
        self.setDaemon(1)
        self.start_orig = self.start
        self.start = self.start_local
        self.frame = None  # to be defined in self.run
        self.lock = threading.Lock()
        self.lock.acquire()  # lock until variables are set
        if autoStart:
            self.start()  # automatically start thread on init

    def run(self):
        import sas.sasview.sasview as sasview
        app = sasview.run_gui()
        self.frame = app.frame

        # define frame and release lock
        # The lock is used to make sure that SetData is defined.
        self.lock.release()

        app.MainLoop()

    def start_local(self):
        self.start_orig()
        # After thread has started, wait until the lock is released
        # before returning so that functions get defined.
        self.lock.acquire()