From fc25ec692910a0c838b7b47ba3e49655150fd961 Mon Sep 17 00:00:00 2001
From: Fabian Joswig <fabian.joswig@ed.ac.uk>
Date: Wed, 8 Dec 2021 22:42:13 +0000
Subject: [PATCH] test: test for Corr.plateau added

---
 tests/correlators_test.py | 8 ++++++++
 1 file changed, 8 insertions(+)

diff --git a/tests/correlators_test.py b/tests/correlators_test.py
index f27bba54..6c3418f8 100644
--- a/tests/correlators_test.py
+++ b/tests/correlators_test.py
@@ -91,6 +91,14 @@ def test_fit_correlator():
     assert fit_res[1] == my_corr[1] - my_corr[0]
 
 
+def test_plateau():
+    my_corr = pe.correlators.Corr([pe.pseudo_Obs(1.01324, 0.05, 't'), pe.pseudo_Obs(1.042345, 0.008, 't')])
+
+    my_corr.plateau([0, 1], method="fit")
+    my_corr.plateau([0, 1], method="mean")
+    with pytest.raises(Exception):
+        my_corr.plateau()
+
 def test_utility():
     corr_content = []
     for t in range(8):