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)) 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)