From a861ce7e7488703e48c4a62e3a61abff835a4c3f Mon Sep 17 00:00:00 2001
From: Dave Grote <grote1@llnl.gov>
Date: Mon, 3 Jul 2023 18:41:09 -0700
Subject: [PATCH] Add tests/test_indextype.py

---
 tests/test_indextype.py | 74 +++++++++++++++++++++++++++++++++++++++++
 1 file changed, 74 insertions(+)
 create mode 100644 tests/test_indextype.py

diff --git a/tests/test_indextype.py b/tests/test_indextype.py
new file mode 100644
index 00000000..cee85cd0
--- /dev/null
+++ b/tests/test_indextype.py
@@ -0,0 +1,74 @@
+# -*- coding: utf-8 -*-
+
+import numpy as np
+import pytest
+
+import amrex.space3d as amr
+
+
+@pytest.mark.skipif(amr.Config.spacedim != 1, reason="Requires AMREX_SPACEDIM = 1")
+def test_indextype_1d():
+    obj = amr.IndexType(amr.IndexType.CellIndex.NODE)
+    assert obj.node_centered()
+    assert not obj.cell_centered()
+    with pytest.raises(IndexError):
+        obj[-2]
+    with pytest.raises(IndexError):
+        obj[1]
+
+
+@pytest.mark.skipif(amr.Config.spacedim != 2, reason="Requires AMREX_SPACEDIM = 2")
+def test_indextype_2d():
+    obj = amr.IndexType(amr.IndexType.CellIndex.NODE, amr.IndexType.CellIndex.CELL)
+    assert obj.node_centered(0)
+    assert obj.cell_centered(1)
+    assert obj.node_centered(-2)
+    assert obj.cell_centered(-1)
+
+    with pytest.raises(IndexError):
+        obj[-3]
+    with pytest.raises(IndexError):
+        obj[2]
+
+
+@pytest.mark.skipif(amr.Config.spacedim != 3, reason="Requires AMREX_SPACEDIM = 3")
+def test_indextype_3d():
+    obj = amr.IndexType(amr.IndexType.CellIndex.NODE, amr.IndexType.CellIndex.CELL, amr.IndexType.CellIndex.NODE)
+
+    # Check indexing
+    assert obj.node_centered(0)
+    assert obj.cell_centered(1)
+    assert obj.node_centered(2)
+    assert obj.node_centered(-3)
+    assert obj.cell_centered(-2)
+    assert obj.node_centered(-1)
+    with pytest.raises(IndexError):
+        obj[-4]
+    with pytest.raises(IndexError):
+        obj[3]
+
+    # Check methods
+    obj.set(1)
+    assert obj.node_centered()
+    obj.unset(1)
+    assert not obj.node_centered()
+
+
+def test_indextype_static():
+    cell = amr.IndexType.cell_type();
+    for i in range(amr.Config.spacedim):
+        assert not cell[i]
+
+    node = amr.IndexType.node_type();
+    for i in range(amr.Config.spacedim):
+        assert node[i]
+
+    assert cell == amr.IndexType.cell_type()
+    assert node == amr.IndexType.node_type()
+    assert cell < node
+
+
+def test_indextype_conversions():
+    node = amr.IndexType.node_type();
+    assert node.ix_type() == amr.IntVect(1)
+    assert node.to_IntVect() == amr.IntVect(1)