Esempio n. 1
0
    def test_BC_attribute_shapes(self):
        # Assert column arrays
        ix = np.array([[120, 121]]).T
        bc = BoundaryCondition(ix, "pressure", 500)
        assert bc.face.shape[0] == 2
        assert bc.face.shape[1] == 1
        print(bc.type)
        assert bc.type.shape[0] == 2
        assert bc.type.shape[1] == 1
        assert bc.value.shape[0] == 2
        assert bc.value.shape[1] == 1

        G = cartGrid([2, 2, 2])
        bc = BoundaryCondition()
        bc.addPressureSide(G, "top", 100)
        assert bc.face.shape[0] == 4
        assert bc.face.shape[1] == 1
        assert bc.type.shape[0] == 4
        assert bc.type.shape[1] == 1
        assert bc.value.shape[0] == 4
        assert bc.value.shape[1] == 1