.gitignore + test_grids.py

EconForge · May 23, 2017 · c3b555c · c3b555c
1 parent 6b06053
commit c3b555c
Show file tree

Hide file tree

Showing 2 changed files with 29 additions and 0 deletions.
diff --git a/.gitignore b/.gitignore
@@ -1 +1,3 @@
 *.pyc
+*.mypy_cache*
+*.ipynb_checkpoints*
diff --git a/dolo/tests/test_grids.py b/dolo/tests/test_grids.py
@@ -0,0 +1,27 @@
+def test_grids():
+
+    from dolo.numeric.grids import CartesianGrid, UnstructuredGrid, NonUniformCartesianGrid, SmolyakGrid
+    from dolo.numeric.grids import n_nodes, nodes, node
+    print("Cartsian Grid")
+    grid = CartesianGrid([0.1, 0.3], [9, 0.4], [50, 10])
+    print(grid.nodes())
+    print(nodes(grid))
+
+    print("UnstructuredGrid")
+    ugrid = UnstructuredGrid([[0.1, 0.3], [9, 0.4], [50, 10]])
+    print(nodes(ugrid))
+    print(node(ugrid,0))
+    print(n_nodes(ugrid))
+
+
+    print("Non Uniform CartesianGrid")
+    ugrid = NonUniformCartesianGrid([[0.1, 0.3], [9, 0.4], [50, 10]])
+    print(nodes(ugrid))
+    print(node(ugrid,0))
+    print(n_nodes(ugrid))
+
+    print("Smolyak Grid")
+    sg = SmolyakGrid([0.1, 0.2], [1.0, 2.0], 2)
+    print(nodes(sg))
+    print(node(sg, 1))
+    print(n_nodes(sg))