def test_4_by_5(): status = sgrid.node_status((4, 5)) assert_equal(status.dtype, np.int8) assert_array_equal( status, np.array([ 1, 1, 1, 1, 1, 1, 0, 0, 0, 1, 1, 0, 0, 0, 1, 1, 1, 1, 1, 1, ]))
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_4_by_5(self): status = sgrid.node_status((4, 5)) self.assertEqual(status.dtype, np.int8) self.assertArrayEqual(status, np.array([1, 1, 1, 1, 1, 1, 0, 0, 0, 1, 1, 0, 0, 0, 1, 1, 1, 1, 1, 1, ]))
def test_with_node_status(): status = sgrid.node_status((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_4_by_5(): status = sgrid.node_status((4, 5)) assert_equal(status.dtype, np.int8) assert_array_equal(status, np.array([1, 1, 1, 1, 1, 1, 0, 0, 0, 1, 1, 0, 0, 0, 1, 1, 1, 1, 1, 1, ]))
def test_no_interiors(): status = sgrid.node_status((2, 3)) assert_equal(status.dtype, np.int8) assert_array_equal(status, np.array([ 1, 1, 1, 1, 1, 1, ]))
def test_no_interiors(): status = sgrid.node_status((2, 3)) assert_equal(status.dtype, np.int8) assert_array_equal(status, np.array([1, 1, 1, 1, 1, 1]))
def test_no_interiors(self): status = sgrid.node_status((2, 3)) self.assertEqual(status.dtype, np.int8) self.assertArrayEqual(status, np.array([1, 1, 1, 1, 1, 1,]))