def test_replace_error_message(self):
"""Test that an exception is raised when replacing layers in
an uninitialized Grid is attempted."""
grid = Grid()
with pytest.raises(
AttributeError,
match=
"Can't replace layer. The Grid has not yet been initialized."):
grid.replace_layer(1, [1, 2, 3])
def test_replace_layer(self, idx, expected_transposed_grid):
"""Test that layer replacement works properly."""
raw_grid = [[0, 3], [1, 4], [2, 5]]
grid = Grid(raw_grid)
grid.replace_layer(idx, [6, 7, 8])
assert np.array_equal(grid.raw_grid.T, expected_transposed_grid)
assert np.array_equal(grid.raw_grid,
_transpose(expected_transposed_grid))
