示例#1
0
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))
示例#2
0
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)))