From 8d8fa82bbebc7535ecc2a1dfc293878f12b885cb Mon Sep 17 00:00:00 2001 From: Fabian Joswig Date: Thu, 23 Dec 2021 14:42:57 +0100 Subject: [PATCH 1/2] feat: Path can now be specified in Corr.dump --- pyerrors/correlators.py | 7 ++++--- 1 file changed, 4 insertions(+), 3 deletions(-) diff --git a/pyerrors/correlators.py b/pyerrors/correlators.py index e2d8648e..e074f95c 100644 --- a/pyerrors/correlators.py +++ b/pyerrors/correlators.py @@ -584,16 +584,17 @@ class Corr: return - def dump(self, filename): + def dump(self, filename, **kwargs): """Dumps the Corr into a pickle file Parameters ---------- filename : str Name of the file + path : str + specifies a custom path for the file (default '.') """ - dump_object(self, filename) - return + dump_object(self, filename, **kwargs) def print(self, range=[0, None]): print(self.__repr__(range)) From 6c7f1f06c9783564ee0d68312b07e77397b60bfd Mon Sep 17 00:00:00 2001 From: Fabian Joswig Date: Thu, 23 Dec 2021 14:56:49 +0100 Subject: [PATCH 2/2] 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)