def test_4_by_5(): active_links = sgrid.active_links((4, 5)) 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_with_status_at_node(): status = sgrid.status_at_node((4, 5)) status[6] = sgrid.CLOSED_BOUNDARY active_links = sgrid.active_links((4, 5), node_status_array=status) assert_array_equal( active_links, np.array([2, 3, 7, 8, 11, 12, 13, 21, 22, 23, 24, 25, 26]) )
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_node_status(self): status = sgrid.node_status((4, 5)) status[6] = sgrid.CLOSED_BOUNDARY active_links = sgrid.active_links((4, 5), node_status_array=status) self.assertArrayEqual(active_links, np.array([2, 3, 7, 8, 11, 12, 13, 21, 22, 23, 24, 25, 26]))
def test_with_active_links(): active_links = sgrid.active_links((4, 5)) active_links = active_links[:-1] link_faces = sgrid.face_at_link((4, 5), actives=active_links) BAD = sgrid.BAD_INDEX_VALUE assert_array_equal( link_faces, np.array([ BAD, 0, 1, 2, BAD, BAD, 3, 4, 5, BAD, BAD, 6, 7, 8, BAD, BAD, BAD, BAD, BAD, 9, 10, 11, 12, 13, 14, 15, BAD, BAD, BAD, BAD, BAD ]))
def test_with_active_links(): active_links = sgrid.active_links((4, 5)) active_links = active_links[:-1] link_faces = sgrid.face_at_link((4, 5), actives=active_links) BAD = sgrid.BAD_INDEX_VALUE assert_array_equal(link_faces, np.array([BAD, 0, 1, 2, BAD, BAD, 3, 4, 5, BAD, BAD, 6, 7, 8, BAD, BAD, BAD, BAD, BAD, 9, 10, 11, 12, 13, 14, 15, BAD, BAD, BAD, BAD, BAD]))
def test_4_by_5(self): active_links = sgrid.active_links((4, 5)) self.assertArrayEqual(active_links, np.array([1, 2, 3, 6, 7, 8, 11, 12, 13, 19, 20, 21, 22, 23, 24, 25, 26])) self.assertEqual(len(active_links), sgrid.active_link_count((4, 5)))