def test_add_boundary_data():
grid = two_triangles()
print(grid.boundaries)
# add the boundary definitions:
grid.boundaries = [
(0, 1),
(0, 2),
(1, 3),
(2, 3),
]
# create a UVar object for boundary conditions:
bnds = UVar('bounds', location='boundary', data=[0, 1, 0, 0, 1])
bnds.attributes["long_name"] = "model boundary conditions"
# wrong size for data
with pytest.raises(ValueError):
grid.add_data(bnds)
# correct data
bnds.data = [0, 1, 0, 0]
grid.add_data(bnds)
assert grid.data['bounds'].name == 'bounds'
assert np.array_equal(grid.data['bounds'].data, [0, 1, 0, 0])
def test_add_boundary_data():
grid = two_triangles()
print(grid.boundaries)
# add the boundary definitions:
grid.boundaries = [(0,1),
(0,2),
(1,3),
(2,3),
]
# create a UVar object for boundary conditions:
bnds = UVar('bounds', location='boundary', data=[0, 1, 0, 0, 1])
bnds.attributes["long_name"] = "model boundary conditions"
# wrong size for data
with pytest.raises(ValueError):
grid.add_data(bnds)
# correct data
bnds.data = [0, 1, 0, 0]
grid.add_data(bnds)
assert grid.data['bounds'].name == 'bounds'
assert np.array_equal( grid.data['bounds'].data, [0, 1, 0, 0] )
def test_add_data():
d = UVar('depth', location='node')
assert d.name == 'depth'
assert np.array_equal(d.data, [])
# Add the data:
d.data = [1.0, 2.0, 3.0, 4.0]
assert np.array_equal(d.data, [1.0, 2.0, 3.0, 4.0])
# Duck type check of nd.ndarray.
d.data *= 2
assert np.array_equal(d.data, [2.0, 4.0, 6.0, 8.0])
