Source code for sciunit.unit_test.base

"""Common imports for many unit tests in this directory"""

import sys

import matplotlib as mpl

OSX = sys.platform == "darwin"
if OSX or "Qt" in mpl.rcParams["backend"]:
    mpl.use("Agg")  # Avoid any problems with Macs or headless displays.


[docs]class SuiteBase(object): """Abstract base class for testing suites and scores"""
[docs] def setUp(self): from sciunit.models.examples import UniformModel from sciunit.tests import RangeTest self.M = UniformModel self.T = RangeTest
[docs] def prep_models_and_tests(self): from sciunit import TestSuite t1 = self.T([2, 3], name="test1") t2 = self.T([5, 6]) m1 = self.M(2, 3) m2 = self.M(5, 6) ts = TestSuite([t1, t2], name="MySuite") return (ts, t1, t2, m1, m2)