Python ensure_type 예제들

예제 #1

파일: test_yield.py 프로젝트: rayjzeng/LibCST

 def test_valid(
     self, node: cst.CSTNode, code: str, position: Optional[CodeRange] = None
 ) -> None:
     self.validate_node(
         node,
         code,
         lambda code: ensure_type(
             ensure_type(parse_statement(code), cst.SimpleStatementLine).body[0],
             cst.Expr,
         ).value,
     )

예제 #2

파일: test_import.py 프로젝트: petersktang/LibCST

 def test_valid(self, **kwargs: Any) -> None:
     self.validate_node(
         parser=lambda code: ensure_type(
             parse_statement(code), cst.SimpleStatementLine
         ).body[0],
         **kwargs,
     )

예제 #3

def _assert_parser(code: str) -> cst.Assert:
    return ensure_type(
        ensure_type(parse_statement(code), cst.SimpleStatementLine).body[0],
        cst.Assert)