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