gridlock/__init__.py

@@ -19,5 +19,5 @@ from .error import GridError as GridError
 from .grid import Grid as Grid
 
 __author__ = 'Jan Petykiewicz'
-__version__ = '1.2'
+__version__ = '1.1'
 version = __version__

gridlock/draw.py

@@ -297,7 +297,7 @@ class GridDrawMixin(GridPosMixin):
         p = numpy.array([[-dimensions[0], +dimensions[1]],
                          [+dimensions[0], +dimensions[1]],
                          [+dimensions[0], -dimensions[1]],
-                         [-dimensions[0], -dimensions[1]]], dtype=float) * 0.5
+                         [-dimensions[0], -dimensions[1]]], dtype=float) / 2.0
         thickness = dimensions[2]
         self.draw_polygon(cell_data, 2, center, p, thickness, foreground)
 

gridlock/examples/ex0.py

@@ -29,9 +29,9 @@ if __name__ == '__main__':
     #         numpy.linspace(-5.5, 5.5, 10)]
 
     half_x = [.25, .5, 0.75, 1, 1.25, 1.5, 2, 2.5, 3, 3.5]
-    xyz3 = [numpy.array([-x for x in half_x[::-1]] + [0] + half_x, dtype=float),
+    xyz3 = [numpy.array([-x for x in half_x[::-1]] + [0] + half_x),
-            numpy.linspace(-5.5, 5.5, 10, dtype=float),
+            numpy.linspace(-5.5, 5.5, 10),
-            numpy.linspace(-5.5, 5.5, 10, dtype=float)]
+            numpy.linspace(-5.5, 5.5, 10)]
     eg = Grid(xyz3)
     egc = eg.allocate(0)
     # eg.draw_slab(Direction.z, 0, 10, 2)

gridlock/grid.py

@@ -94,7 +94,7 @@ class Grid(GridDrawMixin, GridReadMixin, GridPosMixin):
         Raises:
             `GridError` on invalid input
         """
-        edge_arrs = [numpy.array(cc) for cc in pixel_edge_coordinates]
+        edge_arrs = [numpy.asarray(cc) for cc in pixel_edge_coordinates]
         self.exyz = [numpy.unique(edges) for edges in edge_arrs]
         self.shifts = numpy.array(shifts, dtype=float)
 

gridlock/test/test_grid.py

@@ -1,4 +1,4 @@
-# import pytest
+import pytest       # type: ignore
 import numpy
 from numpy.testing import assert_allclose       #, assert_array_equal