summaryrefslogtreecommitdiff
path: root/tests/test_pandas.py
diff options
context:
space:
mode:
Diffstat (limited to 'tests/test_pandas.py')
-rw-r--r--tests/test_pandas.py24
1 files changed, 24 insertions, 0 deletions
diff --git a/tests/test_pandas.py b/tests/test_pandas.py
new file mode 100644
index 0000000..b89db6a
--- /dev/null
+++ b/tests/test_pandas.py
@@ -0,0 +1,24 @@
+"""Tests for using data in pandas.[DataFrame|Series]."""
+import os
+
+import numpy as np
+import pytest
+
+import lmfit
+
+pandas = pytest.importorskip('pandas')
+
+
+def test_pandas_guess_from_peak():
+ """Regression test for failure in guess_from_peak with pandas (GH #629)."""
+ data = pandas.read_csv(os.path.join(os.path.dirname(__file__), '..',
+ 'examples', 'peak.csv'))
+ xdat, ydat = np.loadtxt(os.path.join(os.path.dirname(__file__), '..',
+ 'examples', 'peak.csv'),
+ unpack=True, skiprows=1, delimiter=',')
+
+ model = lmfit.models.LorentzianModel()
+ guess_pd = model.guess(data['y'], x=data['x'])
+ guess = model.guess(ydat, x=xdat)
+
+ assert guess_pd == guess