Plotting func/peak finder

cmeutils/plot_tools.py

@@ -0,0 +1,20 @@
+import scipy.signal as signal
+
+
+def find_peaks(data):
+    """Finds peaks in 1-D data.
+    The peaks are any value greater than the average of all the data values. The data values are 1-D values (like x, y, or z) that     you wish to find the peaks of. 
+    Any points/values that excede the max height will be identified as peaks.
+
+    Parameters
+    ----------
+    data : numpy.ndarray, shape (N,1)
+       Such as x, y, OR z.
+    Returns
+    ----------
+     tuple   
+       """
+    total_peaks = signal.find_peaks(data)
+    avg_peaks = sum(data) / len(data)
+    peaks = signal.find_peaks(data, height=avg_peaks)
+    return peaks 

cmeutils/tests/test_dynamics.py

@@ -1,6 +1,6 @@
 import pytest
 
-from cmeutils.tests.base_test import BaseTest
+from base_test import BaseTest
 from cmeutils.dynamics import msd_from_gsd
 
 class TestDynamics(BaseTest):

cmeutils/tests/test_plot.py

@@ -0,0 +1,17 @@
+import pytest
+
+import scipy.signal as signal
+import numpy as np
+
+from base_test import BaseTest
+
+from cmeutils.plot_tools import find_peaks
+
+class TestPlot(BaseTest):
+    def test_plot(self):
+        x = [1,2,3,4,5]
+        data = np.array(x)
+        peaks = signal.find_peaks(data)
+        assert isinstance(data, np.ndarray)
+        assert isinstance(peaks, tuple)
+

cmeutils/tests/test_structure.py

@@ -5,7 +5,7 @@
 import numpy as np
 import freud
 
-from cmeutils.tests.base_test import BaseTest
+from base_test import BaseTest
 
 from cmeutils.structure import gsd_rdf, get_quaternions, order_parameter, all_atom_rdf, get_centers