a b;@srddlmZddlmmZddlTddlTddlTGdddeZGdddeZ e dkrnddl Z e dS) ) CythonTestN)*c@s<eZdZddZddZddZddZd d Zd d Zd S)TestBufferParsingcsfddjS)Ncs SNfragmentsselfrp/Users/vegardjervell/Documents/master/model/venv/lib/python3.9/site-packages/Cython/Compiler/Tests/TestBuffer.pyz)TestBufferParsing.parse..)Zshould_not_failroot)r r rr r parseszTestBufferParsing.parsecs*fddtj}||jdS)Ncs Srrrr rr r rz1TestBufferParsing.not_parseable..)Z should_failErrors CompileError assertEqual message_only)r Zexpected_errorr err r not_parseableszTestBufferParsing.not_parseablecCs<|d}|jdj}|t|t|dt|jdS)Nz(cdef object[float, 4, ndim=2, foo=foo] xr) rstats base_type assertTrue isinstanceTemplatedTypeNoderlenZpositional_args)r tZbufnoderrr test_basics  zTestBufferParsing.test_basiccCs|ddS)Nz$cdef object[short unsigned int, 3] xrr rrr test_type_posszTestBufferParsing.test_type_poscCs|ddS)Nz0cdef object[foo=foo, dtype=short unsigned int] xr r!rrr test_type_keyword sz#TestBufferParsing.test_type_keywordcCs|dddS)Nz%Non-keyword arg following keyword argzcdef object[foo=1, 2] x)rr!rrr test_pos_after_key#sz$TestBufferParsing.test_pos_after_keyN) __name__ __module__ __qualname__rrrr"r#r$rrrr rs rc@sFeZdZddZdddZddZdd Zd d Zd d ZddZ dS)TestBufferOptionscCs||_||jdSr)errorr expect_error)r r)rrr nonfatal_error.sz TestBufferOptions.nonfatal_errorFcCs|dks Jd|}||_|j|t|t|gdj}|s|jdjjd}t|tsZJ|j }| t|t | t|j t |d|j j|S| t|jdjjdkdS)Nzdef f(): cdef object[%s] xZpipelinerobject)r*r NormalizeTree PostParserrbodyrZ CVarDefNoderrrZbase_type_nodeCSimpleBaseTypeNodernamer)r optsr*r rZvardefbuftyperrr parse_opts3s zTestBufferOptions.parse_optscCs"|j|dd|||jjdS)NT)r*)r6rr)r)r Z expected_errr4rrr non_parseCszTestBufferOptions.non_parsecCsL|d}|t|jt||jjdko6|jjdk|d|jdS)Nzunsigned short int, 3r r6rrZ dtype_noder2signedZlongnessrndimr bufrrr Z __test_basicHs zTestBufferOptions.__test_basiccCsL|d}|t|jt||jjdko6|jjdk|d|jdS)Nz ndim=3, dtype=unsigned short intrr8r9r:r=rrr Z __test_dictNs zTestBufferOptions.__test_dictcCs&|d|td|tddS)Nzint, 2zint, 'a'zint, -34)r6r7Z ERR_BUF_NDIMr!rrr Z __test_ndimTs  zTestBufferOptions.__test_ndimcCsZ|jdt|t|gdj}|jdjj}||djjdk||djjdkdS)Nz DEF ndim = 3 def f(): cdef object[int, ndim] x cdef object[ndim=ndim, dtype=int] y r-rr9) rr/r0rrr1rrr<)r rrrrr Z__test_use_DEFYs z TestBufferOptions.__test_use_DEFN)F) r%r&r'r+r6r7Z_TestBufferOptions__test_basicZ_TestBufferOptions__test_dictZ_TestBufferOptions__test_ndimZ _TestBufferOptions__test_use_DEFrrrr r(+s r(__main__) ZCython.TestUtilsrZCython.Compiler.ErrorsZCompilerrZCython.Compiler.NodesZ#Cython.Compiler.ParseTreeTransformsZCython.Compiler.Bufferrr(r%Zunittestmainrrrr s #;