From 6c7f1f06c9783564ee0d68312b07e77397b60bfd Mon Sep 17 00:00:00 2001 From: Fabian Joswig Date: Thu, 23 Dec 2021 14:56:49 +0100 Subject: [PATCH] test: mpm and corr dump test extended --- tests/correlators_test.py | 1 + tests/mpm_test.py | 14 ++++++++++++++ 2 files changed, 15 insertions(+) create mode 100644 tests/mpm_test.py diff --git a/tests/correlators_test.py b/tests/correlators_test.py index 0f43c8d6..16719337 100644 --- a/tests/correlators_test.py +++ b/tests/correlators_test.py @@ -118,6 +118,7 @@ def test_utility(): corr.print([2, 4]) corr.show() + corr.dump('test_dump', path='.') corr.dump('test_dump') new_corr = pe.load_object('test_dump.p') os.remove('test_dump.p') diff --git a/tests/mpm_test.py b/tests/mpm_test.py new file mode 100644 index 00000000..165f09bc --- /dev/null +++ b/tests/mpm_test.py @@ -0,0 +1,14 @@ +import numpy as np +import pyerrors as pe +import pytest + +np.random.seed(0) + + +def test_mpm(): + corr_content = [] + for t in range(8): + f = 0.8 * np.exp(-0.4 * t) + corr_content.append(pe.pseudo_Obs(np.random.normal(f, 1e-2 * f), 1e-2 * f, 't')) + + res = pe.mpm.matrix_pencil_method(corr_content)