From 31a98f81c47bec304a7c4b5541c238e701ffc334 Mon Sep 17 00:00:00 2001 From: Fabian Joswig Date: Thu, 10 Feb 2022 11:03:59 +0000 Subject: [PATCH] tests: openQCD tests split up in two functions, coverage improved --- pyerrors/input/openQCD.py | 4 ++-- tests/openQCD_in_test.py | 40 +++++++++++++++++++++------------------ 2 files changed, 24 insertions(+), 20 deletions(-) diff --git a/pyerrors/input/openQCD.py b/pyerrors/input/openQCD.py index 78d2805d..1c875c1c 100644 --- a/pyerrors/input/openQCD.py +++ b/pyerrors/input/openQCD.py @@ -467,7 +467,7 @@ def extract_t0(path, prefix, dtr_read, xmin, spatial_extent, fit_range=5, **kwar ax1.set_ylabel('Residuals') ax1.set_xlabel(r'$t/a^2$') - plt.show() + plt.draw() return -fit_result[0] / fit_result[1] @@ -504,7 +504,7 @@ def _read_array_openQCD2(fp): elif size == 16: types = 'dd' else: - print('Type not known!') + raise Exception("Type for size '" + str(size) + "' not known.") m = n[0] for i in range(1, d): m *= n[i] diff --git a/tests/openQCD_in_test.py b/tests/openQCD_in_test.py index 81b2e983..18750a07 100644 --- a/tests/openQCD_in_test.py +++ b/tests/openQCD_in_test.py @@ -4,7 +4,7 @@ import pyerrors as pe import pytest -def test_openqcd(): +def test_rwms(): path = './tests//data/openqcd_test/' prefix = 'sfqcd' postfix = '.rwms' @@ -45,10 +45,29 @@ def test_openqcd(): assert(rwfo[1].value == 1.184681251089919) repname = list(rwfo[0].idl.keys())[0] assert(rwfo[0].idl[repname] == range(1, 10)) - rwfo = pe.input.openQCD.read_rwms(path, prefix, version='2.0', files=files, names=names, r_start=[1], r_stop=[8]) + rwfo = pe.input.openQCD.read_rwms(path, prefix, version='2.0', files=files, names=names, r_start=[1], r_stop=[8], print_err=True) assert(rwfo[0].idl[repname] == range(1, 9)) - # Qtop + # t0 + prefix = 'openqcd' + + t0 = pe.input.openQCD.extract_t0(path, prefix, dtr_read=3, xmin=0, spatial_extent=4) + files = ['openqcd2r1.ms.dat'] + names = ['openqcd2|r1'] + t0 = pe.input.openQCD.extract_t0(path, '', dtr_read=3, xmin=0, spatial_extent=4, files=files, names=names, fit_range=2) + t0 = pe.input.openQCD.extract_t0(path, prefix, dtr_read=3, xmin=0, spatial_extent=4, r_start=[1]) + repname = list(rwfo[0].idl.keys())[0] + assert(t0.idl[repname] == range(1, 10)) + t0 = pe.input.openQCD.extract_t0(path, prefix, dtr_read=3, xmin=0, spatial_extent=4, r_start=[2], r_stop=[8]) + repname = list(rwfo[0].idl.keys())[0] + assert(t0.idl[repname] == range(2, 9)) + t0 = pe.input.openQCD.extract_t0(path, prefix, dtr_read=3, xmin=0, spatial_extent=4, fit_range=2, plaquette=True, assume_thermalization=True) + + pe.input.openQCD.extract_t0(path, '', dtr_read=3, xmin=0, spatial_extent=4, files=files, names=names, fit_range=2, plot_fit=True) + +def test_Qtop(): + path = './tests//data/openqcd_test/' + prefix = 'sfqcd' qtop = pe.input.openQCD.read_qtop(path, prefix, c=0.3, version='sfqcd') repname = list(qtop.idl.keys())[0] @@ -78,18 +97,3 @@ def test_openqcd(): qs = pe.input.openQCD.read_qtop_sector(path, '', 0.3, target=0, Zeuthen_flow=True, version='sfqcd') assert((pe.input.openQCD.qtop_projection(qi, target=0) - qs).is_zero()) - - # t0 - - prefix = 'openqcd' - t0 = pe.input.openQCD.extract_t0(path, prefix, dtr_read=3, xmin=0, spatial_extent=4) - files = ['openqcd2r1.ms.dat'] - names = ['openqcd2|r1'] - t0 = pe.input.openQCD.extract_t0(path, '', dtr_read=3, xmin=0, spatial_extent=4, files=files, names=names, fit_range=2) - t0 = pe.input.openQCD.extract_t0(path, prefix, dtr_read=3, xmin=0, spatial_extent=4, r_start=[1]) - repname = list(rwfo[0].idl.keys())[0] - assert(t0.idl[repname] == range(1, 10)) - t0 = pe.input.openQCD.extract_t0(path, prefix, dtr_read=3, xmin=0, spatial_extent=4, r_start=[2], r_stop=[8]) - repname = list(rwfo[0].idl.keys())[0] - assert(t0.idl[repname] == range(2, 9)) - t0 = pe.input.openQCD.extract_t0(path, prefix, dtr_read=3, xmin=0, spatial_extent=4, fit_range=2, plaquette=True, assume_thermalization=True)