fixed gaussian plotting and added tests

tests/test_analysis.py

@@ -6,7 +6,7 @@ import cristallina.analysis
 __author__ = "Alexander Steppke"
 
 	
-def test_2d_gaussian():
+def test_minimal_2d_gaussian():
 
     image = np.array([[0,0,0,0,0],
             [0,0,0,0,0],
@@ -20,3 +20,21 @@ def test_2d_gaussian():
     assert np.allclose(center_y, 2.0, rtol=1e-04)
 
 
+
+def test_2d_gaussian():
+
+        # define normalized 2D gaussian
+        def gauss2d(x=0, y=0, mx=0, my=0, sx=1, sy=1):
+            return (1 / (2 * np.pi * sx * sy) * np.exp(-((x - mx) ** 2 / (2 * sx**2.0) + (y - my) ** 2 / (2 * sy**2))))
+
+        x = np.arange(0, 150, 1)
+        y = np.arange(0, 100, 1)
+        x, y = np.meshgrid(x, y)  
+
+        z = gauss2d(x, y, mx=40, my=50, sx=20, sy=40)
+        
+        center_x, center_y, result = cristallina.analysis.fit_2d_gaussian(z)
+        assert np.allclose(center_x, 40, rtol=1e-04)
+        assert np.allclose(center_y, 50, rtol=1e-04)
+
+