def test_east_links():
links = sgrid.east_links((3, 4))
assert_array_equal(links, np.array([[8, 9, 10, -1], [11, 12, 13, -1], [14, 15, 16, -1]]))
links = sgrid.east_links((1, 4))
assert_array_equal(links, np.array([[0, 1, 2, -1]]))
links = sgrid.east_links((4, 1))
assert_array_equal(links, np.array([[-1, -1, -1, -1]]).T)
def test_east_links():
links = sgrid.east_links((3, 4))
assert_array_equal(
links, np.array([[0, 1, 2, -1], [7, 8, 9, -1], [14, 15, 16, -1]]))
links = sgrid.east_links((1, 4))
assert_array_equal(links, np.array([[0, 1, 2, -1]]))
links = sgrid.east_links((4, 1))
assert_array_equal(links, np.array([[-1, -1, -1, -1]]).T)
def test_east_links(self):
links = sgrid.east_links((3, 4))
self.assertArrayEqual(links,
np.array([[ 8, 9, 10, -1],
[11, 12, 13, -1],
[14, 15, 16, -1]]))
links = sgrid.east_links((1, 4))
self.assertArrayEqual(links, np.array([[0, 1, 2, -1]]))
links = sgrid.east_links((4, 1))
self.assertArrayEqual(links, np.array([[-1, -1, -1, -1]]).T)
