"""Unit tests for sciunit utility functions and classes"""
import unittest
import tempfile
[docs]class UtilsTestCase(unittest.TestCase):
"""Unit tests for sciunit.utils"""
[docs] def test_warnings_traceback(self):
from sciunit.utils import set_warnings_traceback, warn_with_traceback
set_warnings_traceback(True)
warn_with_traceback("This is a test warning", Warning, "utils_tests.py", 13)
set_warnings_traceback(False)
warn_with_traceback("This is a test warning", Warning, "utils_tests.py", 16)
[docs] def test_notebook(self):
from sciunit.utils import NotebookTools
notebookObj = NotebookTools()
notebookObj.execute_notebook("../docs/chapter1")
[docs] def test_log(self):
from sciunit.utils import log, html_log, strip_html
strip_html("test log1")
html_log("test log1", "test log2")
log("Lorem Ipsum")
[docs] def test_assert_dimensionless(self):
import quantities as pq
from sciunit.utils import assert_dimensionless
assert_dimensionless(3 * pq.s * pq.Hz)
try:
assert_dimensionless(3 * pq.s)
except TypeError:
pass
else:
raise Exception("Should have produced a type error")
[docs] def test_dict_hash(self):
from sciunit.base import SciUnit
from sciunit.utils import dict_hash
d1 = {"a": 1, "b": 2, "c": 3}
d2 = {"c": 3, "a": 1, "b": 2}
dh1 = dict_hash(d1)
dh2 = dict_hash(d2)
self.assertTrue(type(dh1) is str)
self.assertTrue(type(dh2) is str)
self.assertEqual(d1, d2)
[docs] def test_import_module_from_path(self):
from sciunit.utils import import_module_from_path
temp_file = tempfile.mkstemp(suffix=".py")[1]
with open(temp_file, "w") as f:
f.write("value = 42")
module = import_module_from_path(temp_file)
self.assertEqual(module.value, 42)
[docs] def test_versioned(self):
from sciunit.models.examples import ConstModel
m = ConstModel(37)
print("Commit hash is %s" % m.version)
print("Remote URL is %s" % m.remote_url)
self.assertTrue("sciunit" in m.remote_url)
[docs] def test_MockDevice(self):
from sciunit.utils import MockDevice
from io import StringIO
s = StringIO()
myMD = MockDevice(s)
myMD.write("test mock device writing")
if __name__ == "__main__":
unittest.main()