test_swangrid.py

import numpy as np
import pytest
import shapely

from rompy.swan import SwanGrid


# test class based on pytest fixtures
@pytest.fixture
def grid():
    x = np.arange(10)
    y = np.arange(10)
    xx, yy = np.meshgrid(x, y)
    return SwanGrid(x=xx, y=yy)


@pytest.fixture
def grid2():
    return SwanGrid(x0=0, y0=0, dx=1, dy=1, nx=10, ny=10)


def test_bbox(grid):
    assert grid.bbox() == [0.0, 0.0, 9.0, 9.0]


def test_boundary(grid):
    bnd = grid.boundary()
    assert isinstance(bnd, shapely.geometry.polygon.Polygon)


def test_grid_shape(grid):
    assert grid.x.shape == (10, 10)
    assert grid.y.shape == (10, 10)


def test_grid_size(grid):
    assert grid.x.size == 100
    assert grid.y.size == 100


def test_grid_ndim(grid):
    assert grid.x.ndim == 2
    assert grid.y.ndim == 2


def test_grid_minmax(grid):
    assert grid.minx == 0
    assert grid.miny == 0
    assert grid.maxx == 9
    assert grid.maxy == 9


def test_initilisation(grid, grid2):
    assert np.array_equal(grid.x, grid2.x)
    assert np.array_equal(grid.y, grid2.y)