Source code for sfepy.tests.test_log

import os
import numpy as nm
import pytest

import sfepy.base.testing as tst

[docs] @pytest.fixture(scope='module') def log_filename(output_dir): return os.path.join(output_dir, 'test_log.txt')
[docs] @pytest.fixture(scope='module') def log(log_filename): from sfepy.base.log import Log log = Log([['x'], ['x^2', 'x^3']], plot_kwargs = [{}, [{'color' : 'b', 'ls' : '', 'marker' : 'o'}, {'color' : 'g', 'ls' : ':', 'marker' : 'x'}]], yscales=['log', 'linear'], xlabels=['x', 'x'], ylabels=['x', 'x^p'], is_plot=False, aggregate=0, sleep=0.0, log_filename=log_filename, formats=[['{:.3e}'], ['{:.5e}'] * 2]) for x in nm.linspace(0, 1, 11): log(x, x**2, x**3, x=[x + 1, x]) if nm.allclose(x, 0.5): log.plot_vlines([0], color='g', linewidth=2) if nm.allclose(x, 0.7): log.plot_vlines([1], color='g', linewidth=2) log(finished=True) return log
[docs] def test_log_rw(log_filename, log, output_dir): from sfepy.base.base import Output from sfepy.base.log import read_log, write_log log, info = read_log(log_filename) filename2 = os.path.join(output_dir, 'test_log2.txt') output = Output('', filename=filename2, quiet=True) write_log(output, log, info) log2, info2 = read_log(filename2) ok = True _ok = info == info2 if not _ok: tst.report('headers are not equal!') tst.report(info) tst.report(info2) ok = ok and _ok for key, val2 in log2.items(): val = log[key] _ok = nm.allclose(val[0], val2[0], rtol=0.0, atol=1e-14) if not _ok: tst.report('x values are not equal!') tst.report(val[0]) tst.report(val2[0]) ok = ok and _ok _ok = nm.allclose(val[1], val2[1], rtol=0.0, atol=1e-14) if not _ok: tst.report('y values are not equal!') tst.report(val[1]) tst.report(val2[1]) ok = ok and _ok _ok = nm.allclose(val[2], val2[2], rtol=0.0, atol=1e-14) if not _ok: tst.report('vlines are not equal!') tst.report(val[2]) tst.report(val2[2]) ok = ok and _ok assert ok