def test_with_link_nodes(): link_nodes = sgrid.node_index_at_link_ends((4, 5)) active_links = sgrid.active_links((4, 5), link_nodes=link_nodes) assert_array_equal(active_links, np.array([1, 2, 3, 6, 7, 8, 11, 12, 13, 19, 20, 21, 22, 23, 24, 25, 26])) assert_equal(len(active_links), sgrid.active_link_count((4, 5)))
def test_with_link_nodes(): link_nodes = sgrid.node_index_at_link_ends((4, 5)) active_links = sgrid.active_links((4, 5), link_nodes=link_nodes) assert_array_equal( active_links, np.array( [1, 2, 3, 6, 7, 8, 11, 12, 13, 19, 20, 21, 22, 23, 24, 25, 26])) assert len(active_links) == sgrid.active_link_count((4, 5))
def test_2d_3_by_2_to_links(): (_, to_indices) = sgrid.node_index_at_link_ends((3, 2)) assert_array_equal(to_indices, np.array([2, 3, 4, 5, 1, 3, 5]))
def test_2d_3_by_2_from_links(): (from_indices, _) = sgrid.node_index_at_link_ends((3, 2)) assert_array_equal(from_indices, np.array([0, 1, 2, 3, 0, 2, 4]))
def test_2d_3_by_2_to_links(): (_, to_indices) = sgrid.node_index_at_link_ends((3, 2)) assert_array_equal(to_indices, np.array([2, 3, 4, 5, 1, 3, 5]))
def test_2d_3_by_2_from_links(): (from_indices, _) = sgrid.node_index_at_link_ends((3, 2)) assert_array_equal(from_indices, np.array([0, 1, 2, 3, 0, 2, 4]))
def test_2d_3_by_2_to_links(self): (_, to_indices) = sgrid.node_index_at_link_ends((3, 2)) self.assertArrayEqual(to_indices, np.array([2, 3, 4, 5, 1, 3, 5]))
def test_2d_3_by_2_from_links(self): (from_indices, _) = sgrid.node_index_at_link_ends((3, 2)) self.assertArrayEqual(from_indices, np.array([0, 1, 2, 3, 0, 2, 4]))