<filename>seaice/shapefiles/test/test_geom.py
import unittest

import numpy as np
import numpy.testing as npt

from seaice.shapefiles.geom import _feature_grid


class Test__feature_grid(unittest.TestCase):
    def test_returns_1s_in_cells_matching_the_value(self):
        grid = np.array([[1, 2],
                         [4, 5]])

        value = 5

        actual = _feature_grid(grid, value)

        expected = np.array([[0, 0],
                             [0, 1]])

        npt.assert_array_equal(expected, actual)

    def test_returns_1s_in_cells_matching_multiple_values(self):
        grid = np.array([[1, 2],
                         [4, 5]])

        values = [4, 5]

        actual = _feature_grid(grid, *values)

        expected = np.array([[0, 0],
                             [1, 1]])

        npt.assert_array_equal(expected, actual)

<filename>seaice/shapefiles/test/test_geom.py
import unittest

import numpy as np
import numpy.testing as npt

from seaice.shapefiles.geom import _feature_grid


class Test__feature_grid(unittest.TestCase):
    def test_returns_1s_in_cells_matching_the_value(self):
        grid = np.array([[1, 2],
                         [4, 5]])

        value = 5

        actual = _feature_grid(grid, value)

        expected = np.array([[0, 0],
                             [0, 1]])

        npt.assert_array_equal(expected, actual)

    def test_returns_1s_in_cells_matching_multiple_values(self):
        grid = np.array([[1, 2],
                         [4, 5]])

        values = [4, 5]

        actual = _feature_grid(grid, *values)

        expected = np.array([[0, 0],
                             [1, 1]])

        npt.assert_array_equal(expected, actual)
