TST: Provide test for utils.point.py

Author: jwboth

tests/unit/test_point.py

@@ -0,0 +1,153 @@
+"""Unit tests for Point class, i.e., Voxel, Coordinate etc.."""
+
+import numpy as np
+
+import darsia
+
+
+def test_make_coordinate():
+    # Single coordinate - Coordinate
+    coord = darsia.Coordinate([1.0, 2.0])
+    assert isinstance(coord, darsia.Coordinate)
+    assert np.allclose(coord, [1.0, 2.0])
+
+    # Single coordinate - make_coordinate
+    coord = darsia.make_coordinate([1.0, 2.0])
+    assert isinstance(coord, darsia.Coordinate)
+    assert np.allclose(coord, [1.0, 2.0])
+
+    # Multiple coordinates - CoordinateArray
+    coords = darsia.CoordinateArray([[1.0, 2.0], [3.0, 4.0]])
+    assert isinstance(coords, darsia.CoordinateArray)
+    assert np.allclose(coords, [[1.0, 2.0], [3.0, 4.0]])
+
+    # Multiple coordinates - make_coordinate
+    coords = darsia.make_coordinate([[1.0, 2.0], [3.0, 4.0]])
+    assert isinstance(coords, darsia.CoordinateArray)
+    assert np.allclose(coords, [[1.0, 2.0], [3.0, 4.0]])
+
+
+def test_make_voxel():
+    # Single voxel - Voxel
+    voxel = darsia.Voxel([1, 2])
+    assert isinstance(voxel, darsia.Voxel)
+    assert np.allclose(voxel, [1, 2])
+
+    # Single voxel - make_voxel
+    voxel = darsia.make_voxel([1, 2])
+    assert isinstance(voxel, darsia.Voxel)
+    assert np.allclose(voxel, [1, 2])
+
+    # Multiple voxels - VoxelArray
+    voxels = darsia.VoxelArray([[1, 2], [3, 4]])
+    assert isinstance(voxels, darsia.VoxelArray)
+    assert np.allclose(voxels, [[1, 2], [3, 4]])
+
+    # Multiple voxels - make_voxel
+    voxels = darsia.make_voxel([[1, 2], [3, 4]])
+    assert isinstance(voxels, darsia.VoxelArray)
+    assert np.allclose(voxels, [[1, 2], [3, 4]])
+
+
+def test_coordinatearray_getitem():
+    # Define CoordinateArray
+    coords = darsia.CoordinateArray([[1.0, 2.0], [3.0, 4.0]])
+
+    # Getitem - single coordinate
+    coord = coords[0]
+    assert isinstance(coord, darsia.Coordinate)
+    assert np.allclose(coord, [1.0, 2.0])
+
+    # Iterator
+    for coord in coords:
+        assert isinstance(coord, darsia.Coordinate)
+
+
+def test_voxelarray_getitem():
+    # Define VoxelArray
+    voxels = darsia.VoxelArray([[1, 2], [3, 4]])
+
+    # Getitem - single voxel
+    voxel = voxels[0]
+    assert isinstance(voxel, darsia.Voxel)
+    assert np.allclose(voxel, [1, 2])
+
+    # Iterator
+    for voxel in voxels:
+        assert isinstance(voxel, darsia.Voxel)
+
+
+def test_point_to_coordinate():
+    # Define reference image
+    image = darsia.Image(
+        img=np.zeros((3, 4), dtype=float), dimensions=[1, 1], space_dim=2
+    )
+
+    # From Coordinate
+    coord = darsia.Coordinate([0.6, 0.2])
+    to_coord = coord.to_coordinate(image.coordinatesystem)
+    assert isinstance(to_coord, darsia.Coordinate)
+    assert np.allclose(to_coord, [0.6, 0.2])
+
+    # From Voxel
+    voxel = darsia.Voxel([2, 1])
+    to_coord = voxel.to_coordinate(image.coordinatesystem)
+    assert isinstance(to_coord, darsia.Coordinate)
+    assert np.allclose(to_coord, [0.25, 1 / 3])
+
+
+def test_point_to_voxel():
+    # Define reference image
+    image = darsia.Image(
+        img=np.zeros((3, 4), dtype=float), dimensions=[1, 1], space_dim=2
+    )
+
+    # From Coordinate
+    coord = darsia.Coordinate([0.6, 0.2])
+    to_voxel = coord.to_voxel(image.coordinatesystem)
+    assert isinstance(to_voxel, darsia.Voxel)
+    assert np.allclose(to_voxel, [2, 2])
+
+    # From Voxel
+    voxel = darsia.Voxel([2, 1])
+    to_voxel = voxel.to_voxel(image.coordinatesystem)
+    assert isinstance(to_voxel, darsia.Voxel)
+    assert np.allclose(to_voxel, [2, 1])
+
+
+def test_point_array_to_coordinate():
+    # Define reference image
+    image = darsia.Image(
+        img=np.zeros((3, 4), dtype=float), dimensions=[1, 1], space_dim=2
+    )
+
+    # From CoordinateArray
+    coords = darsia.CoordinateArray([[0.6, 0.2], [0.1, 0.9]])
+    to_coords = coords.to_coordinate(image.coordinatesystem)
+    assert isinstance(to_coords, darsia.CoordinateArray)
+    assert np.allclose(to_coords, [[0.6, 0.2], [0.1, 0.9]])
+
+    # From VoxelArray
+    voxels = darsia.VoxelArray([[2, 1], [1, 3]])
+    to_coords = voxels.to_coordinate(image.coordinatesystem)
+    assert isinstance(to_coords, darsia.CoordinateArray)
+    assert np.allclose(to_coords, [[0.25, 1 / 3], [0.75, 2 / 3]])
+
+
+def test_point_array_to_voxel():
+    # Define reference image
+    image = darsia.Image(
+        img=np.zeros((3, 4), dtype=float), dimensions=[1, 1], space_dim=2
+    )
+
+    # From CoordinateArray
+    coords = darsia.CoordinateArray([[0.6, 0.2], [0.1, 0.9]])
+    to_voxels = coords.to_voxel(image.coordinatesystem)
+    assert isinstance(to_voxels, darsia.VoxelArray)
+    assert np.allclose(to_voxels, [[2, 2], [0, 0]])
+
+    # From VoxelArray
+    voxels = darsia.VoxelArray([[2, 1], [1, 3]])
+    to_voxels = voxels.to_voxel(image.coordinatesystem)
+    assert isinstance(to_voxels, darsia.VoxelArray)
+    assert np.allclose(to_voxels, [[2, 1], [1, 3]])