U
    ºmœd1  ã                   @   s  d dl Z d dl mZmZ d dlmZ d dlZd dlZd dlZd dlZd dl	Z	d dl
Z
d dlZd dlZd dlmZ d dlmZ d dlmZ d dlmZ d dlmZ d d	lmZ e ¡ d
krÂe d¡ dd„ ZdZdZdZdZdZ dZ!dZ"dZ#dZ$dZ%dZ&dZ'dZ(dddddd d!d"d#d$d%d&gZ)dd'd(dd)d!d"d#d$d%d&gZ*d*d+d&gZ+d,Z,d-Z-d.Z.G d/d0„ d0eƒZ/G d1d2„ d2e/ƒZ0G d3d4„ d4e/ƒZ1G d5d6„ d6e/ƒZ2G d7d8„ d8e/ƒZ3G d9d:„ d:e4ƒZ5G d;d<„ d<e5ƒZ6G d=d>„ d>e/e6ƒZ7G d?d@„ d@e/ƒZ8G dAdB„ dBe/ƒZ9G dCdD„ dDe/ƒZ:G dEdF„ dFe/ƒZ;G dGdH„ dHe/ƒZ<G dIdJ„ dJe4ƒZ=G dKdL„ dLe/e=ƒZ>G dMdN„ dNe/e=ƒZ?G dOdP„ dPe/e=ƒZ@G dQdR„ dRe/ƒZAG dSdT„ dTe/ƒZBG dUdV„ dVe/ƒZCG dWdX„ dXe7ƒZDG dYdZ„ dZe/ƒZEe Fe ¡  Gd[¡d\¡G d]d^„ d^e/ƒƒZHG d_d`„ d`e/ƒZIG dadb„ dbe/ƒZJG dcdd„ dde/ƒZKeLdekre M¡  dS )fé    N)Ú	CFUNCTYPEÚc_int)Úfind_library)Úcontextmanager)Úmkstemp)Úir)Úbinding)Úffi)ÚTestCaseÚarmv7lzlibgcc_s.so.1c               
   C   s^   t  t j¡} z<zt  t jd¡ W n t jk
r<   Y W ¢dS X W ¢dS W 5 t  t j| ¡ X d S )NÚde_DETF)ÚlocaleÚ	setlocaleÚLC_ALLÚError)Úcur© r   úT/home/sam/Atlas/atlas_env/lib/python3.8/site-packages/llvmlite/tests/test_binding.pyÚno_de_locale   s    r   a±  
    ; ModuleID = '<string>'
    source_filename = "asm_sum.c"
    target triple = "{triple}"
    %struct.glob_type = type {{ i64, [2 x i64]}}

    @glob = global i32 0
    @glob_b = global i8 0
    @glob_f = global float 1.5
    @glob_struct = global %struct.glob_type {{i64 0, [2 x i64] [i64 0, i64 0]}}

    define i32 @sum(i32 %.1, i32 %.2) {{
      %.3 = add i32 %.1, %.2
      %.4 = add i32 0, %.3
      ret i32 %.4
    }}
    z 
    ; ModuleID = '<string>'
    target triple = "{triple}"

    define i32 @sum(i32 %.1, i32 %.2) {{
      %.3 = add i32 %.1, %.2
      ret i32 %.3
    }}
    z½
    ; ModuleID = '<string>'
    target triple = "{triple}"
    @mul_glob = global i32 0

    define i32 @mul(i32 %.1, i32 %.2) {{
      %.3 = mul i32 %.1, %.2
      ret i32 %.3
    }}
    z¡
    ; ModuleID = '<string>'
    target triple = "{triple}"

    define i32 @sum(i32 %.1, i32 %.2) {{
      %.3 = fadd i32 %.1, %.2
      ret i32 %.3
    }}
    z
    ; ModuleID = '<string>'
    target triple = "{triple}"

    define void @sum() {{
      %.bug = add i32 1, %.bug
      ret void
    }}
    zh
    ; ModuleID = '<string>'
    target triple = "{triple}"

    declare i32 @sum(i32 %.1, i32 %.2)
    z–
    ; ModuleID = '<string>'
    target triple = "{triple}"

    define void @foo() {{
      %const = fadd double 0.0, 3.14
      ret void
    }}
    zœ
    ; ModuleID = '<string>'
    target triple = "{triple}"

    define void @foo() {{
      call void asm sideeffect "nop", ""()
      ret void
    }}
    a  
    ; ModuleID = '<string>'
    target triple = "{triple}"

    define void @inlineme() {{
        ret void
    }}

    define i32 @caller(i32 %.1, i32 %.2) {{
    entry:
      %stack = alloca i32
      store i32 %.1, i32* %stack
      br label %main
    main:
      %loaded = load i32, i32* %stack
      %.3 = add i32 %loaded, %.2
      %.4 = add i32 0, %.3
      call void @inlineme()
      ret i32 %.4
    }}
aÉ  
; ModuleID = 'test.c'
source_filename = "test.c"
target triple = "{triple}"

; Function Attrs: noinline nounwind optnone ssp uwtable
define void @inlineme() noinline !dbg !15 {{
  ret void, !dbg !18
}}

; Function Attrs: noinline nounwind optnone ssp uwtable
define i32 @foo(i32 %0, i32 %1) !dbg !19 {{
  %3 = alloca i32, align 4
  %4 = alloca i32, align 4
  store i32 %0, i32* %3, align 4
  call void @llvm.dbg.declare(metadata i32* %3, metadata !23, metadata !DIExpression()), !dbg !24
  store i32 %1, i32* %4, align 4
  call void @llvm.dbg.declare(metadata i32* %4, metadata !25, metadata !DIExpression()), !dbg !26
  call void @inlineme(), !dbg !27
  %5 = load i32, i32* %3, align 4, !dbg !28
  %6 = load i32, i32* %4, align 4, !dbg !29
  %7 = add nsw i32 %5, %6, !dbg !30
  ret i32 %7, !dbg !31
}}

; Function Attrs: nofree nosync nounwind readnone speculatable willreturn
declare void @llvm.dbg.declare(metadata, metadata, metadata) #1

attributes #1 = {{ nofree nosync nounwind readnone speculatable willreturn }}

!llvm.module.flags = !{{!1, !2, !3, !4, !5, !6, !7, !8, !9, !10}}
!llvm.dbg.cu = !{{!11}}
!llvm.ident = !{{!14}}

!0 = !{{i32 2, !"SDK Version", [2 x i32] [i32 12, i32 3]}}
!1 = !{{i32 7, !"Dwarf Version", i32 4}}
!2 = !{{i32 2, !"Debug Info Version", i32 3}}
!3 = !{{i32 1, !"wchar_size", i32 4}}
!4 = !{{i32 1, !"branch-target-enforcement", i32 0}}
!5 = !{{i32 1, !"sign-return-address", i32 0}}
!6 = !{{i32 1, !"sign-return-address-all", i32 0}}
!7 = !{{i32 1, !"sign-return-address-with-bkey", i32 0}}
!8 = !{{i32 7, !"PIC Level", i32 2}}
!9 = !{{i32 7, !"uwtable", i32 1}}
!10 = !{{i32 7, !"frame-pointer", i32 1}}
!11 = distinct !DICompileUnit(language: DW_LANG_C99, file: !12, producer: "Apple clang version 13.1.6 (clang-1316.0.21.2.3)", isOptimized: false, runtimeVersion: 0, emissionKind: FullDebug, enums: !13, splitDebugInlining: false, nameTableKind: None, sysroot: "/Applications/Xcode.app/Contents/Developer/Platforms/MacOSX.platform/Developer/SDKs/MacOSX.sdk", sdk: "MacOSX.sdk")
!12 = !DIFile(filename: "test.c", directory: "/")
!13 = !{{}}
!14 = !{{!"Apple clang version 13.1.6 (clang-1316.0.21.2.3)"}}
!15 = distinct !DISubprogram(name: "inlineme", scope: !12, file: !12, line: 1, type: !16, scopeLine: 1, flags: DIFlagPrototyped, spFlags: DISPFlagDefinition, unit: !11, retainedNodes: !13)
!16 = !DISubroutineType(types: !17)
!17 = !{{null}}
!18 = !DILocation(line: 1, column: 22, scope: !15)
!19 = distinct !DISubprogram(name: "foo", scope: !12, file: !12, line: 3, type: !20, scopeLine: 3, flags: DIFlagPrototyped, spFlags: DISPFlagDefinition, unit: !11, retainedNodes: !13)
!20 = !DISubroutineType(types: !21)
!21 = !{{!22, !22, !22}}
!22 = !DIBasicType(name: "int", size: 32, encoding: DW_ATE_signed)
!23 = !DILocalVariable(name: "a", arg: 1, scope: !19, file: !12, line: 3, type: !22)
!24 = !DILocation(line: 3, column: 13, scope: !19)
!25 = !DILocalVariable(name: "b", arg: 2, scope: !19, file: !12, line: 3, type: !22)
!26 = !DILocation(line: 3, column: 20, scope: !19)
!27 = !DILocation(line: 4, column: 5, scope: !19)
!28 = !DILocation(line: 5, column: 12, scope: !19)
!29 = !DILocation(line: 5, column: 16, scope: !19)
!30 = !DILocation(line: 5, column: 14, scope: !19)
!31 = !DILocation(line: 5, column: 5, scope: !19)
aN  
; ModuleID = "<string>"
target triple = "{triple}"

define double @licm(i32 %0) {{
  %2 = alloca i32, align 4
  %3 = alloca double, align 8
  %4 = alloca i32, align 4
  %5 = alloca double, align 8
  store i32 %0, i32* %2, align 4
  store double 0.000000e+00, double* %3, align 8
  store i32 0, i32* %4, align 4
  br label %6

6:                                                ; preds = %14, %1
  %7 = load i32, i32* %4, align 4
  %8 = load i32, i32* %2, align 4
  %9 = icmp slt i32 %7, %8
  br i1 %9, label %10, label %17

10:                                               ; preds = %6
  store double 7.000000e+00, double* %5, align 8
  %11 = load double, double* %5, align 8
  %12 = load double, double* %3, align 8
  %13 = fadd double %12, %11
  store double %13, double* %3, align 8
  br label %14

14:                                               ; preds = %10
  %15 = load i32, i32* %4, align 4
  %16 = add nsw i32 %15, 1
  store i32 %16, i32* %4, align 4
  br label %6

17:                                               ; preds = %6
  %18 = load double, double* %3, align 8
  ret double %18
}}
a€  
    ; ModuleID = "<string>"
    target triple = "{triple}"

    @A = global i32 undef

    define void @ctor_A()
    {{
      store i32 10, i32* @A
      ret void
    }}

    define void @dtor_A()
    {{
      store i32 20, i32* @A
      ret void
    }}

    define i32 @foo()
    {{
      %.2 = load i32, i32* @A
      %.3 = add i32 %.2, 2
      ret i32 %.3
    }}

    @llvm.global_ctors = appending global [1 x {{i32, void ()*, i8*}}] [{{i32, void ()*, i8*}} {{i32 0, void ()* @ctor_A, i8* null}}]
    @llvm.global_dtors = appending global [1 x {{i32, void ()*, i8*}}] [{{i32, void ()*, i8*}} {{i32 0, void ()* @dtor_A, i8* null}}]
    z…; ModuleID = ""
target triple = "unknown-unknown-unknown"
target datalayout = ""

define i32 @"foo"()
{
"<>!*''#":
  ret i32 12345
}
zaddi	sp, sp, -16zsw	a1, 8(sp)zsw	a2, 12(sp)zfld	ft0, 8(sp)zfmv.w.x	ft1, a0zfcvt.d.s	ft1, ft1zfadd.d	ft0, ft1, ft0zfsd	ft0, 8(sp)zlw	a0, 8(sp)zlw	a1, 12(sp)zaddi	sp, sp, 16Úretzsw	a0, 8(sp)zsw	a1, 12(sp)zfcvt.d.s	ft1, fa0zfcvt.d.s	ft0, fa0zfadd.d	fa0, ft0, fa1zc
declare void @a_readonly_func(i8 *) readonly

declare i8* @a_arg0_return_func(i8* returned, i32*)
A@  7f454c4602010100000000000000000001003e000100000000000000000000000000000000000000e0000000000000000000000040000000000040000500010048c1e2204809c24889d048c1c03d4831d048b90120000480001070480fafc8000000000000000000000000000000000000000000000000002f0000000400f1ff00000000000000000000000000000000070000001200020000000000000000001f00000000000000002e74657874005f5f617279626f002e6e6f74652e474e552d737461636b002e737472746162002e73796d746162003c737472696e673e00000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000001f0000000300000000000000000000000000000000000000a800000000000000380000000000000000000000000000000100000000000000000000000000000001000000010000000600000000000000000000000000000040000000000000001f000000000000000000000000000000100000000000000000000000000000000f00000001000000000000000000000000000000000000005f00000000000000000000000000000000000000000000000100000000000000000000000000000027000000020000000000000000000000000000000000000060000000000000004800000000000000010000000200000008000000000000001800000000000000Z>48c1e2204809c24889d048c1c03d4831d048b90120000480001070480fafc8c                   @   s<   e Zd Zdd„ Zdd„ Zedfdd„Zdd	d
„Zdd„ ZdS )ÚBaseTestc                 C   sB   t  ¡  t  ¡  t  ¡  t ¡  tjd d … | _g tjd d …< d S ©N)ÚllvmÚ
initializeZinitialize_native_targetZinitialize_native_asmprinterÚgcÚcollectÚgarbageÚold_garbage©Úselfr   r   r   ÚsetUp…  s    zBaseTest.setUpc                 C   s   t  ¡  |  t jg ¡ | `d S r   )r   r   ÚassertEqualr   r   r   r   r   r   ÚtearDown  s    zBaseTest.tearDownNc                 C   s    |j t ¡ d}t ||¡}|S )N)Útriple)Úformatr   Úget_default_tripleÚparse_assembly)r   ÚasmÚcontextÚmodr   r   r   Úmodule–  s    zBaseTest.moduleÚglobc                 C   s   |d kr|   ¡ }| |¡S r   )r*   Úget_global_variable)r   Únamer)   r   r   r   r+   ›  s    zBaseTest.globc                C   s   t j ¡ }|j|dS )N©Újit)r   ÚTargetÚfrom_default_tripleÚcreate_target_machine)r   r/   Útargetr   r   r   Útarget_machine   s    
zBaseTest.target_machine)r+   N)	Ú__name__Ú
__module__Ú__qualname__r    r"   Úasm_sumr*   r+   r4   r   r   r   r   r   ƒ  s
   	
r   c                   @   s@   e Zd ZdZe ej d¡d¡e e	j
 d¡d¡dd„ ƒƒZdS )	ÚTestDependencieszB
    Test DLL dependencies are within a certain expected set.
    ÚlinuxzLinux-specific testZLLVMLITE_DIST_TESTzDistribution-specific testc              
   C   sR  t jj}tj ¡ }d|d< tjdd|gtj|d}| 	¡ \}}|  
d|j¡ t d¡}tƒ }| ¡  ¡ D ]^}| ¡ }	|	rh|	d dkrh|	d	 }
| |
¡}t|	ƒd
ks¦|s¶|  d|f ¡ | | d	¡¡ qhd|ksØd|krè|  d|f ¡ tdddddddddg	ƒ}t ¡ dkr| d¡ |D ].}
|
 d¡s|
|kr|  d|
|f ¡ qd S )NÚCÚLANGÚobjdumpz-p)ÚstdoutÚenvr   z$^([-_a-zA-Z0-9]+)\.so(?:\.\d+){0,3}$ZNEEDEDé   é   zinvalid NEEDED line: %rÚlibcZ
libpthreadz#failed parsing dependencies? got %rZlibrtZlibdlZlibzÚlibmZlibgcc_szld-linuxZld64ÚPyPyZlibtinfoz	ld-linux-zunexpected dependency %r in %r)r	   ÚlibÚ_nameÚosÚenvironÚcopyÚ
subprocessÚPopenÚPIPEÚcommunicater!   Ú
returncodeÚreÚcompileÚsetÚdecodeÚ
splitlinesÚsplitÚmatchÚlenZfailÚaddÚgroupÚplatformÚpython_implementationÚ
startswith)r   Zlib_pathr?   ÚpÚoutÚ_Zlib_patÚdepsÚlineÚpartsÚdepÚmÚallowedr   r   r   Ú
test_linuxª  s@    
 ÿ

   ÿ
zTestDependencies.test_linuxN)r5   r6   r7   Ú__doc__ÚunittestÚ
skipUnlessÚsysrY   r[   rG   rH   Úgetre   r   r   r   r   r9   ¥  s   ÿÿr9   c                       s\   e Zd ZdZdZ‡ fdd„Zdd„ Zdd„ Zd	d
„ Zdd„ Z	dd„ Z
dd„ Zdd„ Z‡  ZS )ÚTestRISCVABIz`
    Test calling convention of floating point arguments of RISC-V
    using different ABI.
    zriscv32-unknown-linuxc                    s   t ƒ  ¡  t ¡  t ¡  d S r   )Úsuperr    r   Zinitialize_all_targetsZinitialize_all_asmprintersr   ©Ú	__class__r   r   r    Ö  s    
zTestRISCVABI.setUpc              
   C   sV   zt j | j¡ W n> tk
rP } z dt|ƒkr<|  d¡ n|‚W 5 d }~X Y nX d S )Nz/No available targets are compatible with triplez(RISCV target unsupported by linked LLVM.)r   r0   Úfrom_tripler#   ÚRuntimeErrorÚstrZskipTest)r   Úer   r   r   Úcheck_riscv_targetÛ  s    zTestRISCVABI.check_riscv_targetc                 K   s   t j | j¡}|jf |ŽS r   )r   r0   ro   r#   r2   )r   ÚkwargZlltargetr   r   r   Úriscv_target_machineä  s    z!TestRISCVABI.riscv_target_machinec                 C   s   t  ¡ }t  ¡ }t  |||f¡}t  ¡ }t j||dd}| ¡ }t  |¡}|j\}}	| 	||¡}
| 
|
|	¡}| |¡ t t|ƒ¡}| ¡  |S )NÚfpadd©r-   )r   Z
DoubleTypeZ	FloatTypeÚFunctionTypeÚModuleÚFunctionÚappend_basic_blockÚ	IRBuilderÚargsZfpextZfaddr   r   r&   rq   Úverify)r   Zf64Zf32Zfntyr*   ÚfuncÚblockZbuilderÚaÚbZarg0ÚresultÚllmodr   r   r   Úfpadd_ll_moduleè  s    


zTestRISCVABI.fpadd_ll_modulec                 C   sD   g }|  ¡ D ]2}| ¡ }| d¡s| d¡s|dks| |¡ q|S )NÚ.rv   Ú )rS   Ústripr[   Úappend)r   r'   Zasm_listr`   Zs_liner   r   r   Úbreak_up_asmù  s    ÿzTestRISCVABI.break_up_asmc                 C   s8   |   ¡  |  ¡ }| jdd}|  |  | |¡¡t¡ d S )Nú+f,+d)Úfeatures)rs   r…   ru   r!   rŠ   Úemit_assemblyÚriscv_asm_ilp32©r   r„   r3   r   r   r   Útest_rv32d_ilp32  s    ÿzTestRISCVABI.test_rv32d_ilp32c                 C   s:   |   ¡  |  ¡ }| jddd}|  |  | |¡¡t¡ d S )Nr‹   Zilp32f©rŒ   Zabiname)rs   r…   ru   r!   rŠ   r   Úriscv_asm_ilp32fr   r   r   r   Útest_rv32d_ilp32f	  s    ÿzTestRISCVABI.test_rv32d_ilp32fc                 C   s:   |   ¡  |  ¡ }| jddd}|  |  | |¡¡t¡ d S )Nr‹   Zilp32dr‘   )rs   r…   ru   r!   rŠ   r   Úriscv_asm_ilp32dr   r   r   r   Útest_rv32d_ilp32d  s    ÿzTestRISCVABI.test_rv32d_ilp32d)r5   r6   r7   rf   r#   r    rs   ru   r…   rŠ   r   r“   r•   Ú__classcell__r   r   rm   r   rk   Ï  s   		rk   c                   @   s–   e Zd ZdZdd„ Zdd„ Zdd„ Zdd	„ Zd
d„ Zdd„ Z	dd„ Z
dd„ Zdd„ Zdd„ Zdd„ Zdd„ Zdd„ Ze eƒ d¡dd„ ƒZdd „ Zd!S )"ÚTestMiscz7
    Test miscellaneous functions in llvm.binding.
    c                 C   s   |   t¡ d S r   )r*   r8   r   r   r   r   Útest_parse_assembly  s    zTestMisc.test_parse_assemblyc              	   C   sF   |   t¡}|  t¡ W 5 Q R X t|jƒ}|  d|¡ |  d|¡ d S )Nzparsing errorzinvalid operand type)ÚassertRaisesrp   r*   Úasm_parse_errorrq   Ú	exceptionÚassertIn)r   ÚcmÚsr   r   r   Útest_parse_assembly_error   s
    
z"TestMisc.test_parse_assembly_errorc                 C   sj   t  ¡ }t  t  d¡g ¡}t  ||d¡}t  |jdd¡}| t  t  d¡d¡¡ t	|ƒ}|  
|t¡ d S )Né    Úfooz<>!*''#rw   i90  )r   ry   rx   ÚIntTyperz   r|   r{   r   ÚConstantrq   r!   Úasm_nonalphanum_blocklabel)r   r)   ÚftÚfnÚbdr'   r   r   r   Útest_nonalphanum_block_name'  s    z$TestMisc.test_nonalphanum_block_namec                 C   s$   t j ¡ }t j ¡ }||ks t‚d S r   )r   r(   Zget_global_contextÚAssertionError)r   Z	gcontext1Z	gcontext2r   r   r   Útest_global_context0  s    

zTestMisc.test_global_contextc                 C   sH   t  dd¡ t  dd¡ t  d¡}|  |d¡ t  d¡}|  |d ¡ d S )NZ__xyzzyiÒ  i.  Z__foobar)r   Z
add_symbolZaddress_of_symbolr!   ÚassertIs)r   Úaddrr   r   r   Útest_dylib_symbols5  s    

zTestMisc.test_dylib_symbolsc                 C   s"   t  ¡ }|  |t¡ |  |¡ d S r   )r   r%   ÚassertIsInstancerq   Ú
assertTrue)r   r#   r   r   r   Útest_get_default_triple=  s    z TestMisc.test_get_default_triplec                 C   sb   dd„ }t  ¡ }t  ¡ }|  |t¡ |  |¡ || d¡d ƒ}|| d¡d ƒ}|  ||¡ d S )Nc                 S   s   | dkrdS | S d S )NZpowerpc64leÚppc64ler   )Úarchr   r   r   Únormalize_ppcD  s    z7TestMisc.test_get_process_triple.<locals>.normalize_ppcú-r   )r   Zget_process_tripler%   r®   rq   r¯   rT   r!   )r   r³   r#   ÚdefaultZdefault_archZtriple_archr   r   r   Útest_get_process_tripleB  s    
z TestMisc.test_get_process_triplec                 C   sâ   t  ¡ }|  |t¡ |  |t j¡ | ¡ D ]*\}}|  |t¡ |  |¡ |  |t¡ q*|  | 	¡ t¡ d}d 
|¡}|  t |d¡¡ |  t |d¡¡ |  t |d¡¡ t|ƒdkrÈ|  | 	¡ d¡ n|  t || 	¡ ¡¡ d S )Nz[+\-][a-zA-Z0-9\._-]+z^({0}|{0}(,{0})*)?$r‡   z+aaz+a,-bbr   )r   Zget_host_cpu_featuresr®   ÚdictZ
FeatureMapÚitemsrq   r¯   ÚboolÚflattenr$   ÚassertIsNotNonerO   rU   rV   r!   )r   rŒ   ÚkÚvZre_termÚregexr   r   r   Útest_get_host_cpu_featuresT  s     

z#TestMisc.test_get_host_cpu_featuresc                 C   s"   t  ¡ }|  |t¡ |  |¡ d S r   )r   Úget_host_cpu_namer®   rq   r¯   )r   Úcpur   r   r   Útest_get_host_cpu_namek  s    zTestMisc.test_get_host_cpu_namec                 C   s   d}t  tjd|g¡ d S )Na2  if 1:
            from llvmlite import binding as llvm

            llvm.initialize()
            llvm.initialize_native_target()
            llvm.initialize_native_asmprinter()
            llvm.initialize_all_targets()
            llvm.initialize_all_asmprinters()
            llvm.shutdown()
            ú-c©rJ   Ú
check_callri   Ú
executable©r   Úcoder   r   r   Útest_initfinip  s    
zTestMisc.test_initfinic                 C   s   d}t  tjd|g¡ d S )Nz„if 1:
            from llvmlite import binding as llvm

            llvm.set_option("progname", "-debug-pass=Disabled")
            rÃ   rÄ   rÇ   r   r   r   Útest_set_option}  s    zTestMisc.test_set_optionc                 C   s:   t j\}}}ddddg}|  |f|¡ |  |tdƒ¡ d S )N)é   )é   )é   )é   é
   )r   Úllvm_version_inforœ   Úrange)r   ÚmajorÚminorÚpatchZvalidr   r   r   Útest_version‡  s    zTestMisc.test_versionc                 C   s   t  ¡  d S r   )r   Zcheck_jit_executionr   r   r   r   Útest_check_jit_executionŽ  s    z!TestMisc.test_check_jit_executionzLocale not availablec                 C   s8   |   t¡}t|ƒ}t tjd¡ t|ƒ}|  ||¡ d S )Nr   )r*   Úasm_double_localerq   r   r   r   r!   )r   rc   ÚexpectÚgotr   r   r   Útest_print_double_locale‘  s
    
z!TestMisc.test_print_double_localec                 C   s$   d}d}t j|d|g}t |¡ d S )Nzfrom llvmlite import bindingz-WerrorrÃ   )ri   rÆ   rJ   rÅ   )r   rÈ   ÚflagsZcmdargsr   r   r   Útest_no_accidental_warnings  s    z$TestMisc.test_no_accidental_warningsN)r5   r6   r7   rf   r˜   rŸ   r¨   rª   r­   r°   r¶   r¿   rÂ   rÉ   rÊ   rÕ   rÖ   rg   ZskipIfr   rÚ   rÜ   r   r   r   r   r—     s"   	

r—   c                   @   s´   e Zd Zdd„ Zdd„ Zdd„ Zdd„ Zd	d
„ Zdd„ Zdd„ Z	dd„ Z
dd„ Zdd„ Zdd„ Zdd„ Zdd„ Zdd„ Zdd„ Zdd „ Zd!d"„ Zd#d$„ Zd%d&„ Zd'd(„ Zd)d*„ Zd+S ),ÚTestModuleRefc                 C   s*   |   ¡ }t|ƒ ¡ }|  | d¡|¡ d S )Nz; ModuleID =)r*   rq   rˆ   r¯   r[   ©r   r)   rž   r   r   r   Útest_str¦  s    zTestModuleRef.test_strc              	   C   sD   |   ¡ }t|ƒ | ¡  |  tj¡ t|ƒ W 5 Q R X | ¡  d S r   )r*   rq   Úcloser™   ÚctypesÚArgumentError©r   r)   r   r   r   Ú
test_close«  s    zTestModuleRef.test_closec              
   C   sr   |   ¡ }t|ƒ | t|ƒ W 5 Q R X |  tj¡ t|ƒ W 5 Q R X |  t¡ | W 5 Q R X W 5 Q R X d S r   )r*   rq   r™   rá   râ   rp   rã   r   r   r   Ú	test_with³  s    zTestModuleRef.test_withc                 C   s4   |   ¡ }d|_|  |jd¡ d|_|  |jd¡ d S )Nr¡   Úbar)r*   r-   r!   rã   r   r   r   Ú	test_name¾  s
    zTestModuleRef.test_namec                 C   s   |   ¡ }|  |jd¡ d S )Nz	asm_sum.c)r*   r!   Zsource_filerã   r   r   r   Útest_source_fileÅ  s    zTestModuleRef.test_source_filec                 C   s2   |   ¡ }|j}|  |t¡ ||_|  ||j¡ d S r   )r*   Zdata_layoutr®   rq   r!   rÞ   r   r   r   Útest_data_layoutÉ  s
    zTestModuleRef.test_data_layoutc                 C   s6   |   ¡ }|j}|  |t ¡ ¡ d|_|  |jd¡ d S )Nr‡   )r*   r#   r!   r   r%   rÞ   r   r   r   Útest_tripleÐ  s
    zTestModuleRef.test_triplec              	   C   sZ   |   ¡ }|  | ¡ d ¡ |   t¡}|  t¡}| ¡  W 5 Q R X t|jƒ}|  d|¡ d S )Nz%.bug = add i32 1, %.bug)	r*   r«   r~   Úasm_verification_failr™   rp   rq   r›   rœ   )r   r)   r   rž   r   r   r   Útest_verify×  s    

zTestModuleRef.test_verifyc              	   C   s^   |   ¡ }| d¡}|  |tj¡ |  |jd¡ |  t¡ | d¡ W 5 Q R X ~t	|j ƒ d S )NÚsumr¡   )
r*   Úget_functionr®   r   ÚValueRefr!   r-   r™   Ú	NameErrorrq   ©r   r)   r¦   r   r   r   Útest_get_functionâ  s    
zTestModuleRef.test_get_functionc              	   C   sZ   |   ¡ }| d¡}|  |jd¡ |  t dt|ƒ¡¡ |  t	¡ | d¡ W 5 Q R X d S )Nzstruct.glob_typeú8%struct\.glob_type(\.[\d]+)? = type { i64, \[2 x i64\] }zstruct.doesnt_exist)
r*   Zget_struct_typer!   r-   r»   rO   rU   rq   r™   rð   )r   r)   Zst_tyr   r   r   Útest_get_struct_typeï  s    
þz"TestModuleRef.test_get_struct_typec              	   C   s^   |   ¡ }| d¡}|  |tj¡ |  |jd¡ |  t¡ | d¡ W 5 Q R X ~t	|j ƒ d S )Nr+   ræ   )
r*   r,   r®   r   rï   r!   r-   r™   rð   rq   )r   r)   Úgvr   r   r   Útest_get_global_variableû  s    
z&TestModuleRef.test_get_global_variablec                 C   sR   |   ¡ }|j}~t|dd„ d}|  t|ƒd¡ |  dd„ |D ƒddd	d
g¡ d S )Nc                 S   s   | j S r   rw   )Úvaluer   r   r   Ú<lambda>  ó    z5TestModuleRef.test_global_variables.<locals>.<lambda>)Úkeyé   c                 S   s   g | ]
}|j ‘qS r   rw   )Ú.0Úgr   r   r   Ú
<listcomp>  s     z7TestModuleRef.test_global_variables.<locals>.<listcomp>r+   Úglob_bZglob_fÚglob_struct)r*   Úglobal_variablesÚsortedr!   rV   )r   r)   ÚitZglobsr   r   r   Útest_global_variables  s    
ÿz#TestModuleRef.test_global_variablesc                 C   s>   |   ¡ }|j}~t|ƒ}|  t|ƒd¡ |  |d jd¡ d S ©Nr@   r   rí   )r*   Ú	functionsÚlistr!   rV   r-   )r   r)   r  Úfuncsr   r   r   Útest_functions  s    zTestModuleRef.test_functionsc                 C   s^   |   ¡ }|j}~t|ƒ}|  t|ƒd¡ |  t d|d j¡¡ |  t dt	|d ƒ¡¡ d S )Nr@   ústruct\.glob_type(\.[\d]+)?r   ró   )
r*   Zstruct_typesr  r!   rV   r»   rO   rU   r-   rq   )r   r)   r  Ústructsr   r   r   Útest_structs  s    
ÿ
þzTestModuleRef.test_structsc              	   C   st   |   ¡ }|   t¡}| |¡ |  tdd„ |jD ƒƒddg¡ | d¡ | ¡  |  t	j
¡ | d¡ W 5 Q R X d S )Nc                 s   s   | ]}|j V  qd S r   rw   ©rü   Úfr   r   r   Ú	<genexpr>*  s     z-TestModuleRef.test_link_in.<locals>.<genexpr>Úmulrí   )r*   Úasm_mulÚlink_inr!   r  r  rî   rà   r™   rá   râ   )r   ÚdestÚsrcr   r   r   Útest_link_in%  s    

 ÿ
zTestModuleRef.test_link_inc                 C   st   |   ¡ }|   t¡}|j|dd |  tdd„ |jD ƒƒddg¡ | ¡  |  tdd„ |jD ƒƒdg¡ | d¡ d S )NT)Zpreservec                 s   s   | ]}|j V  qd S r   rw   r  r   r   r   r  5  s     z6TestModuleRef.test_link_in_preserve.<locals>.<genexpr>r  rí   c                 s   s   | ]}|j V  qd S r   rw   r  r   r   r   r  7  s     )r*   r  r  r!   r  r  rà   rî   )r   r  Zsrc2r   r   r   Útest_link_in_preserve0  s    
 ÿz#TestModuleRef.test_link_in_preservec              	   C   sH   |   ¡ }|   t¡}|  t¡}| |¡ W 5 Q R X |  dt|jƒ¡ d S )Nzsymbol multiply defined)r*   Úasm_sum2r™   rp   r  rœ   rq   r›   )r   r  r  r   r   r   r   Útest_link_in_error:  s
    
z TestModuleRef.test_link_in_errorc                 C   s6   |   ¡ }| ¡ }d}d}|  | |¡p.| |¡¡ d S )Ns   ÞÀs   BC)r*   Ú
as_bitcoder¯   r[   )r   r)   ÚbcZbitcode_wrapper_magicZbitcode_magicr   r   r   Útest_as_bitcodeC  s    ÿzTestModuleRef.test_as_bitcodec              	   C   sj   |   t¡}t d¡ W 5 Q R X |  dt|jƒ¡ tjd dk rT|  dt|jƒ¡ n|  dt|jƒ¡ d S )Nrù   zLLVM bitcode parsing errorr   é	   zInvalid bitcode signaturez(file too small to contain bitcode header)r™   rp   r   Úparse_bitcoderœ   rq   r›   rÐ   )r   r   r   r   r   Útest_parse_bitcode_errorM  s     ÿz&TestModuleRef.test_parse_bitcode_errorc                 C   sT   t  ¡ }| j|d ¡ }t  ¡ }t  ||¡}|  | ¡ |¡ | d¡ | d¡ d S )N)r(   rí   r+   )r   Zcreate_contextr*   r  r  r!   rî   r,   )r   Zcontext1r  Zcontext2r)   r   r   r   Útest_bitcode_roundtripY  s    
z$TestModuleRef.test_bitcode_roundtripc                 C   s4   |   ¡ }| ¡ }|  ||¡ |  | ¡ | ¡ ¡ d S r   )r*   ÚcloneZassertIsNotr!   r  )r   rc   Zclonedr   r   r   Útest_cloningd  s    zTestModuleRef.test_cloningN)r5   r6   r7   rß   rä   rå   rç   rè   ré   rê   rì   rò   rô   rö   r  r	  r  r  r  r  r  r  r  r!  r   r   r   r   rÝ   ¤  s*   	
	
rÝ   c                   @   s‚   e Zd ZdZd dd„Zdd„ Zdd„ Zd	d
„ Zdd„ Zdd„ Z	dd„ Z
dd„ Zdd„ Zdd„ Zdd„ Zdd„ Zdd„ Zdd„ ZdS )!ÚJITTestMixinz*
    Mixin for ExecutionEngine tests.
    rí   c                 C   s,   |  ¡  | |¡}|  |¡ ttttƒ|ƒS r   )Úfinalize_objectÚget_function_addressr¯   r   r   )r   ÚeeÚ	func_nameZcfptrr   r   r   Úget_sump  s    

zJITTestMixin.get_sumc              	   C   sB   |   ¡ }|  |¡&}|  |¡}|ddƒ}|  d|¡ W 5 Q R X d S )NrA   éûÿÿÿéýÿÿÿ)r*   r/   r'  r!   )r   r)   r%  ÚcfuncÚresr   r   r   Útest_run_codev  s
    

zJITTestMixin.test_run_codec              	   C   sB   |   |  ¡ ¡}| ¡  | ¡  |  tj¡ | ¡  W 5 Q R X d S r   )r/   r*   rà   r™   rá   râ   r#  ©r   r%  r   r   r   rä   }  s
    zJITTestMixin.test_closec              
   C   sh   |   |  ¡ ¡}| W 5 Q R X |  t¡ | W 5 Q R X W 5 Q R X |  tj¡ | ¡  W 5 Q R X d S r   )r/   r*   r™   rp   rá   râ   r#  r-  r   r   r   rå   „  s    
zJITTestMixin.test_withc                 C   s&   |   ¡ }|  |¡}| ¡  | ¡  d S r   ©r*   r/   rà   ©r   r)   r%  r   r   r   Útest_module_lifetimeŽ  s    
z!JITTestMixin.test_module_lifetimec                 C   s&   |   ¡ }|  |¡}| ¡  | ¡  d S r   r.  r/  r   r   r   Útest_module_lifetime2”  s    
z"JITTestMixin.test_module_lifetime2c              	   C   sf   |   |  ¡ ¡}|  t¡}| |¡ |  t¡ | |¡ W 5 Q R X |  |j¡ | ¡  |  	|j¡ d S r   )
r/   r*   r  Ú
add_moduler™   ÚKeyErrorÚassertFalseÚclosedrà   r¯   ©r   r%  r)   r   r   r   Útest_add_moduleš  s    

zJITTestMixin.test_add_modulec                 C   s6   |   |  ¡ ¡}|  t¡}| |¡ | ¡  | ¡  d S r   ©r/   r*   r  r2  rà   r6  r   r   r   Útest_add_module_lifetime¤  s
    

z%JITTestMixin.test_add_module_lifetimec                 C   s6   |   |  ¡ ¡}|  t¡}| |¡ | ¡  | ¡  d S r   r8  r6  r   r   r   Útest_add_module_lifetime2«  s
    

z&JITTestMixin.test_add_module_lifetime2c              	   C   sp   |   |  ¡ ¡}|  t¡}| |¡ | |¡ |  t¡ | |¡ W 5 Q R X |  |j¡ | 	¡  |  |j¡ d S r   )
r/   r*   r  r2  Zremove_moduler™   r3  r4  r5  rà   r6  r   r   r   Útest_remove_module²  s    


zJITTestMixin.test_remove_modulec                 C   s>   |   ¡ }|  |¡}|j}|  |j|¡ t|ƒ ~~t|ƒ d S r   )r*   r/   Útarget_datar«   rq   )r   r)   r%  Útdr   r   r   Útest_target_data½  s    
zJITTestMixin.test_target_datac           	      C   sè   |   ¡ }|  |¡}|j}| d¡}| d¡}| d¡}tjdk rDdnd}|||fD ]}|  | |j¡|¡ qR|  | 	|j¡d¡ |  | 
|j¡d¡ |  | 	|j¡d¡ |  | 
|j¡d¡ |  | 	|j¡d	¡ |  | 
|j¡d
¡ d S )Nr+   rÿ   r   ì        rû   é   r@   )r@   rA   rû   é   )rû   r@  )r*   r/   r<  r,   ri   Úmaxsizer!   Úget_abi_sizeÚtypeÚget_pointee_abi_sizeZget_pointee_abi_alignmentrœ   )	r   r)   r%  r=  Úgv_i32Zgv_i8Z	gv_structÚpointer_sizerý   r   r   r   Útest_target_data_abi_enquiriesÇ  s    



z+JITTestMixin.test_target_data_abi_enquiriesc                    sð   g ‰ ‡ fdd„}|   ¡ }|  |¡}| |¡ |  tˆ ƒd¡ |  |¡}|ddƒ |  tˆ ƒd¡ |  ˆ d d |¡ |  ˆ d d t¡ g ˆ d d …< |   t	¡}| 
|¡ |  |d¡}|  tˆ ƒd¡ |  ˆ d d |¡ |  ˆ d d t¡ d S )Nc                    s   ˆ   | |f¡ d S r   ©r‰   ©r)   Úbuf©Únotifiesr   r   Únotifyß  s    z5JITTestMixin.test_object_cache_notify.<locals>.notifyr   rA   r(  r@   r  )r*   r/   Úset_object_cacher!   rV   r'  r«   r®   Úbytesr  r2  )r   rN  r)   r%  r*  Zmod2r   rL  r   Útest_object_cache_notifyÜ  s$    





z%JITTestMixin.test_object_cache_notifyc                    sJ  g ‰g ‰ ‡fdd„}‡ fdd„}|   ¡ }|  |¡}| ||¡ |  tˆƒd¡ |  tˆ ƒd¡ |  |¡}|  tˆƒd¡ |  tˆ ƒd¡ |  ˆ d |¡ ˆd d ‰‡ ‡fdd„}g ˆd d …< g ˆ d d …< |   t¡}|  |¡}| ||¡ |  tˆƒd¡ |  tˆ ƒd¡ |  |¡}|  |d	d
ƒd¡ |  tˆƒd¡ |  tˆ ƒd¡ d S )Nc                    s   ˆ   | |f¡ d S r   rI  rJ  rL  r   r   rN  û  s    z8JITTestMixin.test_object_cache_getbuffer.<locals>.notifyc                    s   ˆ   | ¡ d S r   rI  ©r)   )Ú
getbuffersr   r   Ú	getbufferþ  s    z;JITTestMixin.test_object_cache_getbuffer.<locals>.getbufferr   r@   c                    s   ˆ   | ¡ ˆS r   rI  rR  )rS  Ú
sum_bufferr   r   Úgetbuffer_successful  s    
zFJITTestMixin.test_object_cache_getbuffer.<locals>.getbuffer_successfulrA   r(  r)  )r*   r/   rO  r!   rV   r'  r«   r  )r   rN  rT  r)   r%  r*  rV  r   )rS  rM  rU  r   Útest_object_cache_getbuffer÷  s4    




z(JITTestMixin.test_object_cache_getbufferN)rí   )r5   r6   r7   rf   r'  r,  rä   rå   r0  r1  r7  r9  r:  r;  r>  rH  rQ  rW  r   r   r   r   r"  k  s   



r"  c                   @   s   e Zd Zdd„ Zdd„ ZdS )ÚJITWithTMTestMixinc                 C   sf   | j dd}|  ¡ }|  ||¡}| |¡}|  d|¡ | d¡ | |¡}|  d|¡ |  ||¡ dS )z%Test TargetMachineRef.emit_assembly()Tr.   rí   N)r4   r*   r/   r   rœ   Zset_asm_verbosityZassertNotEqual)r   r4   r)   r%  Zraw_asmZraw_asm_verboser   r   r   Útest_emit_assembly&  s    


z%JITWithTMTestMixin.test_emit_assemblyc                 C   sZ   | j dd}|  ¡ }|  ||¡}| |¡}|  |t¡ tj d¡rV|  	d|dd… ¡ dS )z#Test TargetMachineRef.emit_object()Tr.   r:   s   ELFNrÏ   )
r4   r*   r/   Úemit_objectr®   rP  ri   rY   r[   rœ   )r   r4   r)   r%  Úcode_objectr   r   r   Útest_emit_object2  s    
z#JITWithTMTestMixin.test_emit_objectN)r5   r6   r7   rY  r\  r   r   r   r   rX  $  s   rX  c                   @   s   e Zd ZdZddd„ZdS )Ú	TestMCJitz@
    Test JIT engines created with create_mcjit_compiler().
    Nc                 C   s    |d kr| j dd}t ||¡S )NTr.   )r4   r   Úcreate_mcjit_compiler)r   r)   r4   r   r   r   r/   C  s    zTestMCJit.jit)N)r5   r6   r7   rf   r/   r   r   r   r   r]  >  s   r]  c                   @   s´   e Zd Zdd„ Zdd„ Zdd„ Zdd„ Zd	d
„ Zdd„ Zdd„ Z	dd„ Z
dd„ Zdd„ Zdd„ Zdd„ Zdd„ Zdd„ Zdd„ Zdd „ Zd!d"„ Zd#d$„ Zd%d&„ Zd'd(„ Zd)d*„ Zd+S ),ÚTestValueRefc                 C   s&   |   ¡ }| d¡}|  t|ƒd¡ d S )Nr+   z@glob = global i32 0)r*   r,   r!   rq   ©r   r)   r+   r   r   r   rß   K  s    
zTestValueRef.test_strc                 C   s8   |   ¡ }| d¡}|  |jd¡ d|_|  |jd¡ d S )Nr+   Úfoobar)r*   r,   r!   r-   r`  r   r   r   rç   P  s
    
zTestValueRef.test_namec                 C   sp   |   ¡ }| d¡}|j}|  |jtj¡ ||_|  |j|¡ dD ]*}||_|  |jtj¡ |  |jj|¡ q@d S )Nr+   )ÚinternalÚexternal)r*   r,   Úlinkager®   r   ZLinkager!   r-   )r   r)   r+   rd  r   r   r   Útest_linkageW  s    
zTestValueRef.test_linkagec                 C   sp   |   ¡ }| d¡}|j}|  |jtj¡ ||_|  |j|¡ dD ]*}||_|  |jtj¡ |  |jj|¡ q@d S )Nr+   )ZhiddenZ	protectedrµ   )r*   r,   Ú
visibilityr®   r   Z
Visibilityr!   r-   )r   r)   r+   rf  r   r   r   Útest_visibilityc  s    
zTestValueRef.test_visibilityc                 C   sp   |   ¡ }| d¡}|j}|  |jtj¡ ||_|  |j|¡ dD ]*}||_|  |jtj¡ |  |jj|¡ q@d S )Nr+   )Z	dllimportZ	dllexportrµ   )r*   r,   Ústorage_classr®   r   ZStorageClassr!   r-   )r   r)   r+   rh  r   r   r   Útest_storage_classo  s    
zTestValueRef.test_storage_classc              	   C   sR   |   ¡ }| d¡}| d¡ |  t¡}| d¡ W 5 Q R X |  t|jƒd¡ d S )Nrí   Z	nocaptureZzextzno such attribute 'zext')r*   rî   Zadd_function_attributer™   Ú
ValueErrorr!   rq   r›   )r   r)   r¦   Zraisesr   r   r   Útest_add_function_attribute{  s    

z(TestValueRef.test_add_function_attributec                 C   s$   |   ¡ }| d¡}|  |j |¡ d S ©Nr+   )r*   r,   r«   r`  r   r   r   Útest_moduleƒ  s    
zTestValueRef.test_modulec                 C   s*   |   ¡ }| d¡}|j}|  |tj¡ d S rl  )r*   r,   rD  r®   r   ZTypeRef©r   r)   r+   Útpr   r   r   Ú	test_typeˆ  s    
zTestValueRef.test_typec                 C   sL   |   ¡ }| d¡}|j}|  |jd¡ | d¡}|  t d|jjj¡¡ d S )Nr+   r‡   r   r
  )	r*   r,   rD  r!   r-   r»   rO   rU   Úelement_type)r   r)   r+   ro  Ústr   r   r   Útest_type_nameŽ  s    


ÿzTestValueRef.test_type_namec                 C   s,   |   ¡ }| d¡}|j}|  t|ƒd¡ d S )Nr+   zi32*)r*   r,   rD  r!   rq   rn  r   r   r   Útest_type_printing_variable—  s    
z(TestValueRef.test_type_printing_variablec                 C   s(   |   ¡ }| d¡}|  t|jƒd¡ d S )Nrí   zi32 (i32, i32)*)r*   rî   r!   rq   rD  rñ   r   r   r   Útest_type_printing_function  s    
z(TestValueRef.test_type_printing_functionc                 C   sV   |   ¡ }| d¡}|  |jj¡ |  t dt|jƒ¡¡ |  t dt|jj	ƒ¡¡ d S )Nr   z%struct\.glob_type(\.[\d]+)?\*ró   )
r*   r,   r¯   rD  Z
is_pointerr»   rO   rU   rq   rq  )r   r)   rr  r   r   r   Útest_type_printing_struct¢  s    

ÿ
þz&TestValueRef.test_type_printing_structc                 C   s   |   ¡ }| ¡  | ¡  d S r   )r+   rà   )r   r+   r   r   r   rä   ¬  s    zTestValueRef.test_closec                 C   s:   |   ¡  d¡}|   t¡ d¡}|  |j¡ |  |j¡ d S )Nrí   )r*   rî   Úasm_sum_declarer4  Zis_declarationr¯   )r   ÚdefinedZdeclaredr   r   r   Útest_is_declaration±  s    z TestValueRef.test_is_declarationc                 C   s>   |   t¡}t|jƒ}|  t|ƒd¡ |D ]}|  |j¡ q(d S ©Nrû   )r*   r8   r  r  r!   rV   r¯   Ú	is_global)r   r)   Úgvarsr½   r   r   r   Útest_module_global_variables·  s
    

z)TestValueRef.test_module_global_variablesc              	   C   sœ   |   ¡ }t|jƒ}|  t|ƒd¡ |d }|  |j¡ |  |jd¡ |  t	¡ |j
 W 5 Q R X |  t	¡ |j W 5 Q R X |  t	¡ |j W 5 Q R X d S r  )r*   r  r  r!   rV   r¯   Úis_functionr-   r™   rj  ÚinstructionsÚoperandsÚopcode)r   r)   r  r   r   r   r   Útest_module_functions¾  s    
z"TestValueRef.test_module_functionsc              	   C   sì   |   ¡ }| d¡}|  |j¡ t|jƒ}|  t|ƒd¡ |  |d j¡ |  |d j¡ |  |d j	d¡ |  t
|d jƒd¡ |  |d j	d¡ |  t
|d jƒd¡ |  t¡ |d j W 5 Q R X |  t¡ |d j W 5 Q R X d S )Nrí   rA   r   r@   ú.1Úi32ú.2)r*   rî   r¯   r~  r  Ú	argumentsr!   rV   Úis_argumentr-   rq   rD  r™   rj  Úblocks)r   r)   r   r}   r   r   r   Útest_function_argumentsÍ  s    

z$TestValueRef.test_function_argumentsc                 C   s@   |   ¡  d¡}t|jƒ}|  t|ƒd¡ |d }|  |j¡ d S )Nrí   r@   r   )r*   rî   r  rˆ  r!   rV   r¯   Zis_block)r   r   rˆ  r€   r   r   r   Útest_function_blocksß  s
    
z!TestValueRef.test_function_blocksc                 C   sœ   |   ¡  d¡}tt|jƒd jƒ}|  t|ƒd¡ |  |d j¡ |  |d j¡ |  |d j¡ |  |d j	d¡ |  |d j	d¡ |  |d j	d¡ d S )Nrí   r   é   r@   rA   rW   r   )
r*   rî   r  rˆ  r  r!   rV   r¯   Zis_instructionr  )r   r   Zinstsr   r   r   Útest_block_instructionsæ  s    z$TestValueRef.test_block_instructionsc                 C   sÂ   |   ¡  d¡}tt|jƒd jƒd }|  |jd¡ t|jƒ}|  t|ƒd¡ |  	|d j
¡ |  	|d j
¡ |  |d jd¡ |  t|d jƒd¡ |  |d jd¡ |  t|d jƒd¡ d S )	Nrí   r   rW   rA   r@   rƒ  r„  r…  )r*   rî   r  rˆ  r  r!   r  r€  rV   r¯   Z
is_operandr-   rq   rD  )r   r   rW   r€  r   r   r   Útest_instruction_operandsñ  s    
z&TestValueRef.test_instruction_operandsc                 C   sŒ   |   t¡}|jD ]v}t|jƒ}|jdkr8|  |dg¡ q|jdkr|  |g ¡ t|jƒ}|  t|d jƒdg¡ |  t|d jƒg ¡ qd S )NZa_readonly_funcs   readonlyZa_arg0_return_funcr   s   returnedr@   )r*   Úasm_attributesr  r  Ú
attributesr-   r!   r†  )r   r)   r   Úattrsr}   r   r   r   Útest_function_attributesþ  s    





z%TestValueRef.test_function_attributesN)r5   r6   r7   rß   rç   re  rg  ri  rk  rm  rp  rs  rt  ru  rv  rä   ry  r}  r‚  r‰  rŠ  rŒ  r  r‘  r   r   r   r   r_  I  s*   	
r_  c                   @   s4   e Zd Zdd„ Zdd„ Zdd„ Zdd„ Zd	d
„ ZdS )Ú
TestTargetc              	   C   sb   t jj}|  t¡}|dƒ W 5 Q R X |  dt|jƒ¡ t  ¡ }||ƒ}|  	|j
|¡ | ¡  d S )Nra  z(No available targets are compatible with)r   r0   ro   r™   rp   rœ   rq   r›   r%   r!   r#   rà   )r   r  r   r#   r3   r   r   r   Útest_from_triple  s    ÿzTestTarget.test_from_triplec                 C   s@   t j t  ¡ ¡}| ddddd¡ t  ¡ }| |dddd¡ d S )Nr‡   r@   rµ   )r   r0   ro   r%   r2   rÀ   )r   r3   rÁ   r   r   r   Útest_create_target_machine  s    z%TestTarget.test_create_target_machinec                 C   s<   t j t  ¡ ¡}t j ¡ }|  |jt¡ |  |j|j¡ d S r   )	r   r0   ro   r%   r1   r®   r-   rq   r!   ©r   ÚtÚur   r   r   rç      s    
zTestTarget.test_namec                 C   s<   t j t  ¡ ¡}t j ¡ }|  |jt¡ |  |j|j¡ d S r   )	r   r0   ro   r%   r1   r®   Údescriptionrq   r!   r•  r   r   r   Útest_description&  s    
zTestTarget.test_descriptionc                 C   s8   t j t  ¡ ¡}t|ƒ}|  |j|¡ |  |j|¡ d S r   )r   r0   ro   r%   rq   rœ   r-   r˜  )r   r3   rž   r   r   r   rß   ,  s    zTestTarget.test_strN)r5   r6   r7   r“  r”  rç   r™  rß   r   r   r   r   r’    s
   r’  c                   @   s,   e Zd Zdd„ Zdd„ Zdd„ Zdd„ Zd	S )
ÚTestTargetDatac                 C   s
   t  d¡S )Nz%e-m:e-i64:64-f80:128-n8:16:32:64-S128)r   Zcreate_target_datar   r   r   r   r<  5  s    zTestTargetData.target_datac                 C   s(   |   ¡ }|  ¡ }|  | |j¡d¡ d S )Nr@  )r<  r+   r!   rC  rD  ©r   r=  r+   r   r   r   Útest_get_abi_size8  s    z TestTargetData.test_get_abi_sizec                 C   sF   |   ¡ }|  ¡ }|  | |j¡d¡ |  d¡}|  | |j¡d¡ d S )Nrû   r   rA  )r<  r+   r!   rE  rD  r›  r   r   r   Útest_get_pointee_abi_size=  s
    
z(TestTargetData.test_get_pointee_abi_sizec              	   C   sj   |   ¡ }|  d¡}|  t¡ | |jd¡ W 5 Q R X |jj}|  | |d¡d¡ |  | |d¡d¡ d S )Nr   r   r@   r@  )r<  r+   r™   rj  Zget_element_offsetrD  rq  r!   )r   r=  r+   Zstruct_typer   r   r   Útest_get_struct_element_offsetF  s    
z-TestTargetData.test_get_struct_element_offsetN)r5   r6   r7   r<  rœ  r  rž  r   r   r   r   rš  3  s   	rš  c                   @   s   e Zd Zdd„ Zdd„ ZdS )ÚTestTargetMachinec                 C   s"   | j dd}t ¡ }| |¡ d S )NFr.   )r4   r   Úcreate_module_pass_managerZadd_analysis_passes)r   ÚtmÚpmr   r   r   Útest_add_analysis_passesT  s    z*TestTargetMachine.test_add_analysis_passesc                 C   sN   | j dd}|j}|  ¡ }| d¡}tjdk r2dnd}|  | |j¡|¡ d S )NFr.   r+   r?  rû   r@  )	r4   r<  r*   r,   ri   rB  r!   rC  rD  )r   r¡  r=  r)   rF  rG  r   r   r   Útest_target_data_from_tmY  s    
z*TestTargetMachine.test_target_data_from_tmN)r5   r6   r7   r£  r¤  r   r   r   r   rŸ  R  s   rŸ  c                   @   sd   e Zd Zdd„ Zdd„ Zdd„ Zdd„ Zd	d
„ Zdd„ Zdd„ Z	dd„ Z
dd„ Zdd„ Zdd„ ZdS )ÚTestPassManagerBuilderc                 C   s   t  ¡ S r   )r   ZPassManagerBuilderr   r   r   r   Úpmbe  s    zTestPassManagerBuilder.pmbc                 C   s   t  ¡ }d|_d|_d S )NrA   r‹  )r   Úcreate_pass_manager_builderÚinlining_thresholdÚ	opt_level©r   r¦  r   r   r   Útest_old_apih  s    z#TestPassManagerBuilder.test_old_apic                 C   s   |   ¡ }| ¡  | ¡  d S r   )r¦  rà   rª  r   r   r   rä   n  s    z!TestPassManagerBuilder.test_closec                 C   s<   |   ¡ }|  |jt¡ tdƒD ]}||_|  |j|¡ qd S rz  )r¦  r®   r©  ÚintrÑ   r!   ©r   r¦  Úir   r   r   Útest_opt_levels  s
    z%TestPassManagerBuilder.test_opt_levelc                 C   s<   |   ¡ }|  |jt¡ tdƒD ]}||_|  |j|¡ qd S rz  )r¦  r®   Z
size_levelr¬  rÑ   r!   r­  r   r   r   Útest_size_levelz  s
    z&TestPassManagerBuilder.test_size_levelc              	   C   s8   |   ¡ }|  t¡ |j W 5 Q R X dD ]
}||_q(d S )N)é   éP   i^  )r¦  r™   ÚNotImplementedErrorr¨  r­  r   r   r   Útest_inlining_threshold  s
    z.TestPassManagerBuilder.test_inlining_thresholdc                 C   s8   |   ¡ }|  |jt¡ dD ]}||_|  |j|¡ qd S ©N)TF)r¦  r®   Zdisable_unroll_loopsr¹   r!   ©r   r¦  r‚   r   r   r   Útest_disable_unroll_loopsˆ  s
    z0TestPassManagerBuilder.test_disable_unroll_loopsc                 C   s8   |   ¡ }|  |jt¡ dD ]}||_|  |j|¡ qd S rµ  )r¦  r®   Zloop_vectorizer¹   r!   r¶  r   r   r   Útest_loop_vectorize  s
    z*TestPassManagerBuilder.test_loop_vectorizec                 C   s8   |   ¡ }|  |jt¡ dD ]}||_|  |j|¡ qd S rµ  )r¦  r®   Zslp_vectorizer¹   r!   r¶  r   r   r   Útest_slp_vectorize–  s
    z)TestPassManagerBuilder.test_slp_vectorizec                 C   s.   |   ¡ }t ¡ }| |¡ | ¡  | ¡  d S r   )r¦  r   r   Úpopulaterà   )r   r¦  r¢  r   r   r   Ú!test_populate_module_pass_manager  s
    
z8TestPassManagerBuilder.test_populate_module_pass_managerc                 C   s8   |   ¡ }|  ¡ }t |¡}| |¡ | ¡  | ¡  d S r   )r*   r¦  r   Úcreate_function_pass_managerrº  rà   )r   r)   r¦  r¢  r   r   r   Ú#test_populate_function_pass_manager¤  s    

z:TestPassManagerBuilder.test_populate_function_pass_managerN)r5   r6   r7   r¦  r«  rä   r¯  r°  r´  r·  r¸  r¹  r»  r½  r   r   r   r   r¥  c  s   r¥  c                   @   s   e Zd Zdd„ Zdd„ ZdS )ÚPassManagerTestMixinc                 C   s   t  ¡ }d|_d|_|S )NrA   i,  )r   r§  r©  r¨  rª  r   r   r   r¦  ¯  s    zPassManagerTestMixin.pmbc                 C   s   |   ¡ }| ¡  | ¡  d S r   )r¢  rà   ©r   r¢  r   r   r   rä   µ  s    zPassManagerTestMixin.test_closeN)r5   r6   r7   r¦  rä   r   r   r   r   r¾  ­  s   r¾  c                   @   s<   e Zd Zdd„ Zdd„ Zdd„ Zdd„ Zd	d
„ Zdd„ ZdS )ÚTestModulePassManagerc                 C   s   t  ¡ S r   ©r   r   r   r   r   r   r¢  ½  s    zTestModulePassManager.pmc                 C   s¼   |   ¡ }|  ¡  |¡ |  ¡ }t|ƒ}| |¡ t|ƒ}| ¡ }t|ƒD ]f\}}| ¡  	d¡rHddh}|D ]}	|	|krj q„qjt
dƒ‚||	hA  ¡ }
|  d|¡ |  |
|¡  q¸qHt
dƒ‚d S )Nzret i32z%.3ú%.4zexpected tokens not foundzexpected IR not found)r¢  r¦  rº  r*   rq   ÚrunrS   Ú	enumeraterˆ   r[   rp   Úpoprœ   ÚassertNotIn)r   r¢  r)   Úorig_asmÚopt_asmZopt_asm_splitÚidxÚlÚtoksr–  Z
othertokenr   r   r   Útest_runÀ  s&    
zTestModulePassManager.test_runc                 C   s^   |   ¡ }| d¡ |  ¡  |¡ |  t¡}| |¡\}}|  |¡ |  d|¡ |  d|¡ d S )NéF   ÚPassedÚinlineme©	r¢  Úadd_function_inlining_passr¦  rº  r*   Úasm_inlineasm2Úrun_with_remarksr¯   rœ   ©r   r¢  r)   ÚstatusÚremarksr   r   r   Ú'test_run_with_remarks_successful_inlineÞ  s    


z=TestModulePassManager.test_run_with_remarks_successful_inlinec                 C   sj   |   ¡ }| d¡ |  ¡  |¡ |  t¡}| |¡\}}|  |¡ |  d|¡ |  d|¡ |  d|¡ d S )Nr   ZMissedrÏ  znoinline function attribute)	r¢  rÑ  r¦  rº  r*   Úasm_inlineasm3rÓ  r¯   rœ   rÔ  r   r   r   Ú#test_run_with_remarks_failed_inlineé  s    


z9TestModulePassManager.test_run_with_remarks_failed_inlinec                 C   sV   |   ¡ }| d¡ |  ¡  |¡ |  t¡}|j|dd\}}|  |¡ |  d|¡ d S )NrÍ  Únothing©Zremarks_filterr‡   )	r¢  rÑ  r¦  rº  r*   rÒ  rÓ  r¯   r!   rÔ  r   r   r   Ú'test_run_with_remarks_inline_filter_outö  s    


z=TestModulePassManager.test_run_with_remarks_inline_filter_outc                 C   sb   |   ¡ }| d¡ |  ¡  |¡ |  t¡}|j|dd\}}|  |¡ |  d|¡ |  d|¡ d S )NrÍ  zinlin.*rÛ  rÎ  rÏ  rÐ  rÔ  r   r   r   Ú&test_run_with_remarks_inline_filter_inÿ  s    


z<TestModulePassManager.test_run_with_remarks_inline_filter_inN)	r5   r6   r7   r¢  rÌ  r×  rÙ  rÜ  rÝ  r   r   r   r   rÀ  »  s   	rÀ  c                   @   s>   e Zd Zddd„Zdd„ Zdd„ Zdd	„ Zd
d„ Zdd„ ZdS )ÚTestFunctionPassManagerNc                 C   s   |p
|   ¡ }t |¡S r   )r*   r   r¼  rã   r   r   r   r¢    s    zTestFunctionPassManager.pmc                 C   s   |   ¡ }| ¡  | ¡  d S r   )r¢  r   Úfinalizer¿  r   r   r   rÉ     s    z%TestFunctionPassManager.test_initfinic                 C   sx   |   ¡ }| d¡}|  |¡}|  ¡  |¡ | ¡  t|ƒ}| ¡  | |¡ | 	¡  t|ƒ}|  
d|¡ |  d|¡ d S )Nrí   rÂ  )r*   rî   r¢  r¦  rº  rà   rq   r   rÃ  rß  rœ   rÆ  )r   r)   r¦   r¢  rÇ  rÈ  r   r   r   rÌ    s    


z TestFunctionPassManager.test_runc                 C   s€   |   t¡}| d¡}|  |¡}| ¡  |  ¡  |¡ | ¡  | ¡  | 	|¡\}}| 
¡  |  |¡ |  d|¡ |  d|¡ d S )NÚlicmrÎ  ©r*   Úlicm_asmrî   r¢  Úadd_licm_passr¦  rº  rà   r   rÓ  rß  r¯   rœ   ©r   r)   r¦   r¢  ÚokrÖ  r   r   r   Útest_run_with_remarks$  s    



z-TestFunctionPassManager.test_run_with_remarksc                 C   sx   |   t¡}| d¡}|  |¡}| ¡  |  ¡  |¡ | ¡  | ¡  |j	|dd\}}| 
¡  |  |¡ |  d|¡ d S )Nrà  rÚ  rÛ  r‡   )r*   râ  rî   r¢  rã  r¦  rº  rà   r   rÓ  rß  r¯   r!   rä  r   r   r   Ú test_run_with_remarks_filter_out3  s    



z8TestFunctionPassManager.test_run_with_remarks_filter_outc                 C   s„   |   t¡}| d¡}|  |¡}| ¡  |  ¡  |¡ | ¡  | ¡  |j	|dd\}}| 
¡  |  |¡ |  d|¡ |  d|¡ d S )Nrà  rÛ  rÎ  rá  rä  r   r   r   Útest_run_with_remarks_filter_inA  s    



z7TestFunctionPassManager.test_run_with_remarks_filter_in)N)	r5   r6   r7   r¢  rÉ   rÌ  ræ  rç  rè  r   r   r   r   rÞ  
  s   
rÞ  c                   @   s   e Zd Zdd„ Zdd„ ZdS )Ú
TestPassesc                 C   s   t  ¡ S r   rÁ  r   r   r   r   r¢  S  s    zTestPasses.pmc                 C   s*  |   ¡ }| ¡  | ¡  | ¡  | d¡ | ¡  | ¡  | ¡  | ¡  | 	¡  | 
¡  | ¡  | ¡  | ¡  | ¡  | ¡  | ¡  | ¡  | ¡  | ¡  | ¡  | ¡  | ¡  | d¡ | ¡  | ¡  | ¡  | ¡  | ¡  | d¡ | ¡  | ¡  |  ¡  | !¡  | "¡  | #¡  | $¡  | %¡  | &¡  | '¡  | (¡  | )¡  | *¡  | +¡  | ,¡  | -¡  | .¡  | /¡  | 0¡  | 1¡  | 2¡  | 3¡  | 4¡  | 5¡  | 6¡  | 7¡  | 8¡  | 9¡  | :¡  | ;¡  | <¡  | =¡  | >¡  | ?¡  | @¡  | A¡  | B¡  | C¡  d S )Néá   é*   é   )Dr¢  Zadd_constant_merge_passZadd_dead_arg_elimination_passZadd_function_attrs_passrÑ  Zadd_global_dce_passZadd_global_optimizer_passZadd_ipsccp_passZadd_dead_code_elimination_passZadd_cfg_simplification_passZadd_gvn_passZadd_instruction_combining_passrã  Zadd_sccp_passZadd_sroa_passZ"add_type_based_alias_analysis_passZadd_basic_alias_analysis_passZadd_loop_rotate_passZadd_region_info_passZadd_scalar_evolution_aa_passZ)add_aggressive_dead_code_elimination_passZadd_aa_eval_passZadd_always_inliner_passZadd_arg_promotion_passZadd_break_critical_edges_passZadd_dead_store_elimination_passZ*add_reverse_post_order_function_attrs_passZ)add_aggressive_instruction_combining_passZadd_internalize_passZadd_jump_threading_passZadd_lcssa_passZadd_loop_deletion_passZadd_loop_extractor_passZadd_single_loop_extractor_passZadd_loop_strength_reduce_passZadd_loop_simplification_passZadd_loop_unroll_passZadd_loop_unroll_and_jam_passZadd_loop_unswitch_passZadd_lower_atomic_passZadd_lower_invoke_passZadd_lower_switch_passZadd_memcpy_optimization_passZadd_merge_functions_passZadd_merge_returns_passZadd_partial_inlining_passZ!add_prune_exception_handling_passZ add_reassociate_expressions_passZ"add_demote_register_to_memory_passZadd_sink_passZadd_strip_symbols_passZadd_strip_dead_debug_info_passZadd_strip_dead_prototypes_passZadd_strip_debug_declare_passZadd_strip_nondebug_symbols_passZadd_tail_call_elimination_passZadd_basic_aa_passZadd_dependence_analysis_passZadd_dot_call_graph_passZadd_dot_cfg_printer_passZadd_dot_dom_printer_passZadd_dot_postdom_printer_passZadd_globals_mod_ref_aa_passZadd_iv_users_passZadd_lazy_value_info_passZadd_lint_passZadd_module_debug_info_passZadd_refprune_passr¿  r   r   r   Útest_populateV  sˆ    


zTestPasses.test_populateN)r5   r6   r7   r¢  rí  r   r   r   r   ré  Q  s   ré  c                   @   s0   e Zd Zdd„ Ze e ¡ dkd¡dd„ ƒZdS )Ú	TestDylibc              	   C   s$   |   t¡ t d¡ W 5 Q R X d S )Nzzzzasdkf;jasd;l)r™   rp   r   Úload_library_permanentlyr   r   r   r   Útest_bad_libraryŸ  s    zTestDylib.test_bad_library)ÚLinuxztest only works on Linuxc                 C   s   t dƒ}t |¡ d S )Nrc   )r   r   rï  )r   rC   r   r   r   Ú	test_libm£  s    zTestDylib.test_libmN)	r5   r6   r7   rð  rg   rh   rY   Úsystemrò  r   r   r   r   rî    s
   ÿrî  c                   @   s$   e Zd Zdd„ Zdd„ Zdd„ ZdS )ÚTestAnalysisc                 C   sj   t  ¡ }t  t  d¡t  d¡t  d¡g¡}t  ||d¡}t  | ¡ ¡}|j\}}| ||¡}| 	|¡ |S )Nr    r¡   )
r   ry   rx   r¢   rz   r|   r{   r}   rW   r   )r   rc   r¥   r¦   r§   ÚxÚyÚzr   r   r   Úbuild_ir_module«  s    "

zTestAnalysis.build_ir_modulec                 C   sJ   |   ¡ }| d¡}t |¡}tj|dd}d}|  ||¡ |  ||¡ d S )Nr¡   F©Z	show_instz%.5 = add i32 %.1, %.2)rø  Z
get_globalr   Úget_function_cfgrœ   rÆ  )r   r)   r¡   Údot_showing_instÚdot_without_instÚinstr   r   r   Útest_get_function_cfg_on_irµ  s    

z(TestAnalysis.test_get_function_cfg_on_irc                 C   s‚   |   ¡  d¡}tj|dd}tj|dd}d}|  ||¡ |  ||¡ d}|  ||¡ |  ||¡ d}|  ||¡ |  ||¡ d S )Nrí   Trù  FZdigraphzCFG for 'sum' functionz%.3 = add i32 %.1, %.2)r*   rî   r   rú  rœ   rÆ  )r   rx  rû  rü  ÚprefixÚfnamerý  r   r   r   Útest_function_cfg_on_llvm_value¾  s    z,TestAnalysis.test_function_cfg_on_llvm_valueN)r5   r6   r7   rø  rþ  r  r   r   r   r   rô  ª  s   
	rô  c                   @   s    e Zd Zedd„ ƒZdd„ ZdS )ÚTestTypeParsingc                 c   s$   t  ¡ }|V  t|ƒ}t |¡ d S r   )r   ry   rq   r   r&   )r   r)   r'   r   r   r   Úcheck_parsingÑ  s    zTestTypeParsing.check_parsingc              	   C   s   |   ¡ 6}t t d¡g¡}t ||d¡}t |dg¡|_W 5 Q R X |   ¡ :}tjt d¡gdd}t ||d¡}t |dg¡|_W 5 Q R X d S )Nr    r¡   r@   T)Úpacked)r  r   ZLiteralStructTyper¢   ÚGlobalVariabler£   Zinitializer)r   r)   Útyprõ   r   r   r   Útest_literal_structÛ  s    

ÿz#TestTypeParsing.test_literal_structN)r5   r6   r7   r   r  r  r   r   r   r   r  Ð  s   
	r  c                   @   s   e Zd Zdd„ ZdS )ÚTestGlobalConstructorsc                 C   sš   |   t¡}|  |¡}| ¡  | ¡  | d¡}t tj¡}t 	||¡}|  
|jjd¡ | d¡}t tj¡|ƒ}|  
|ƒ d¡ | ¡  |  
|jjd¡ d S )NÚArÏ   r¡   rÌ   é   )r*   Úasm_global_ctorsr/   r#  Zrun_static_constructorsZget_global_value_addressrá   ZPOINTERZc_int32Úcastr!   Úcontentsr÷   r$  r   Zrun_static_destructors)r   r)   r%  Zptr_addrZptr_tZptrZfoo_addrr¡   r   r   r   Útest_global_ctors_dtorsí  s    



z.TestGlobalConstructors.test_global_ctors_dtorsN)r5   r6   r7   r  r   r   r   r   r  ì  s   r  c                   @   sn   e Zd Zddd„Zdd„ Zdd„ Zdd	„ Zd
d„ Zdd„ Zdd„ Z	dd„ Z
dd„ Zdd„ Zdd„ Zdd„ ZdS )ÚTestGlobalVariablesTc                 C   sZ   t  ¡ }t  d¡}t  ||d¡}||_t|ƒ}|r@|  d|¡ n|  d|¡ |  |¡ d S )Nr    r¡   Zundef)	r   ry   r¢   r  rd  rq   rœ   rÆ  r*   )r   rd  Ú	has_undefr)   r  rõ   r'   r   r   r   Úcheck_global_variable_linkage  s    
z1TestGlobalVariables.check_global_variable_linkagec                 C   s   |   d¡ d S )Nrb  ©r  r   r   r   r   Útest_internal_linkage  s    z)TestGlobalVariables.test_internal_linkagec                 C   s   |   d¡ d S )NÚcommonr  r   r   r   r   Útest_common_linkage  s    z'TestGlobalVariables.test_common_linkagec                 C   s   | j ddd d S )Nrc  F©r  r  r   r   r   r   Útest_external_linkage!  s    z)TestGlobalVariables.test_external_linkagec                 C   s   |   d¡ d S )NZavailable_externallyr  r   r   r   r   Ú!test_available_externally_linkage$  s    z5TestGlobalVariables.test_available_externally_linkagec                 C   s   |   d¡ d S )NÚprivater  r   r   r   r   Útest_private_linkage'  s    z(TestGlobalVariables.test_private_linkagec                 C   s   |   d¡ d S )NZlinkoncer  r   r   r   r   Útest_linkonce_linkage*  s    z)TestGlobalVariables.test_linkonce_linkagec                 C   s   |   d¡ d S )NZweakr  r   r   r   r   Útest_weak_linkage-  s    z%TestGlobalVariables.test_weak_linkagec                 C   s   |   d¡ d S )NZ	appendingr  r   r   r   r   Útest_appending_linkage0  s    z*TestGlobalVariables.test_appending_linkagec                 C   s   | j ddd d S )NZextern_weakFr  r  r   r   r   r   Útest_extern_weak_linkage3  s    z,TestGlobalVariables.test_extern_weak_linkagec                 C   s   |   d¡ d S )NZlinkonce_odrr  r   r   r   r   Útest_linkonce_odr_linkage6  s    z-TestGlobalVariables.test_linkonce_odr_linkagec                 C   s   |   d¡ d S )NZweak_odrr  r   r   r   r   Útest_weak_odr_linkage9  s    z)TestGlobalVariables.test_weak_odr_linkageN)T)r5   r6   r7   r  r  r  r  r  r  r  r  r  r  r  r   r   r   r   r   r    s   
r  Úx86zonly on x86c                   @   s   e Zd Zdd„ ZdS )ÚTestInlineAsmc                 C   s:   t  ¡  | jtd}| jdd}| |¡}|  d|¡ d S )N)r'   Fr.   Znop)r   Zinitialize_native_asmparserr*   Úasm_inlineasmr4   r   rœ   )r   rc   r¡  r'   r   r   r   Útest_inlineasm?  s
    
zTestInlineAsm.test_inlineasmN)r5   r6   r7   r$  r   r   r   r   r"  =  s   r"  c                   @   s0   e Zd ZdZdd„ Zdd„ Zdd„ Zdd	„ Zd
S )ÚTestObjectFilea,  
        ;ModuleID = <string>
        target triple = "{triple}"

        declare i32 @sum(i32 %.1, i32 %.2)

        define i32 @sum_twice(i32 %.1, i32 %.2) {{
            %.3 = call i32 @sum(i32 %.1, i32 %.2)
            %.4 = call i32 @sum(i32 %.3, i32 %.3)
            ret i32 %.4
        }}
    c                 C   s¼   | j dd}|  ¡ }| |¡}tj |¡}d}d}| ¡ D ]r}| ¡ r:d}|  | 	¡ ¡ |  
| ¡ dk¡ |  
t| ¡ ƒdk¡ |  | ¡ ¡ |  
|| ¡ k ¡ | ¡ } q®q:|  
|¡ d S )NFr.   éÿÿÿÿTr   )r4   r*   rZ  r   ÚObjectFileRefÚ	from_dataÚsectionsÚis_textr»   r-   r¯   ÚsizerV   ÚdataÚaddress)r   r4   r)   Úobj_binÚobjZhas_textZlast_addressrž   r   r   r   Útest_object_fileV  s"    
zTestObjectFile.test_object_filec                 C   st   | j dd}|  ¡ }| |¡}tj |¡}t |  | j¡|¡}| |¡ t	t
t
t
ƒ| d¡ƒ}|  |ddƒd¡ d S )NFr.   Ú	sum_twicerA   r‹  rÏ   )r4   r*   rZ  r   r'  r(  r^  Úmod_asmÚadd_object_filer   r   r$  r!   )r   r4   r)   r.  r/  r/   r1  r   r   r   Útest_add_object_filej  s    
ÿ

ÿz#TestObjectFile.test_add_object_filec           	   	   C   s²   | j dd}|  ¡ }| |¡}tƒ \}}zPz"t |d¡}| |¡ | 	¡  W 5 | ¡  X t
 |  | j¡|¡}| |¡ W 5 t |¡ X ttttƒ| d¡ƒ}|  |ddƒd¡ d S )NFr.   Úwbr1  rA   r‹  rÏ   )r4   r*   rZ  r   rG   Úunlinkrà   ÚfdopenÚwriteÚflushr   r^  r2  r3  r   r   r$  r!   )	r   r4   r)   r.  Z	temp_descZ	temp_pathr  r/   r1  r   r   r   Ú$test_add_object_file_from_filesystemz  s&    



ÿ
ÿz3TestObjectFile.test_add_object_file_from_filesystemc                 C   sX   t  t¡}tj |¡}| ¡ D ]4}| ¡ r|  t	| 
¡ ƒd¡ |  | 
¡  ¡ t¡ qd S )Né   )rP  ÚfromhexÚissue_632_elfr   r'  r(  r)  r*  r!   rV   r,  ÚhexÚissue_632_text)r   Úelfr/  rž   r   r   r   Útest_get_section_content”  s    
z'TestObjectFile.test_get_section_contentN)r5   r6   r7   r2  r0  r4  r:  rA  r   r   r   r   r%  G  s
   r%  c                   @   s   e Zd Zdd„ Zdd„ ZdS )ÚTestTimePassesc              	   C   s   t  ¡ }t  ¡ }d|_| |¡ z8t  d¡ | |  ¡ ¡ | |  ¡ ¡ | |  ¡ ¡ W 5 t  ¡ }t  d¡ X |  	|t
¡ |  | d¡d¡ d S )Nr‹  FTzPass execution timing reportr@   )r   r   r§  r©  rº  Úreport_and_reset_timingsZset_time_passesrÃ  r*   r®   rq   r!   Úcount)r   Úmpr¦  Úreportr   r   r   Útest_reporting   s    

zTestTimePasses.test_reportingc                 C   s   |   t ¡ ¡ d S r   )r4  r   rC  r   r   r   r   Útest_empty_report³  s    z TestTimePasses.test_empty_reportN)r5   r6   r7   rG  rH  r   r   r   r   rB  Ÿ  s   rB  c                   @   s   e Zd Zdd„ ZdS )ÚTestLLVMLockCallbacksc              	      sŠ   g ‰ ‡ fdd„}‡ fdd„}t j ||¡ |  ˆ ¡ t  ¡  |  dˆ ¡ |  dˆ ¡ t j ||¡ |  t¡ t j ||¡ W 5 Q R X d S )Nc                      s   ˆ   d¡ d S )NÚacqrI  r   ©Úeventsr   r   rJ  ¼  s    z6TestLLVMLockCallbacks.test_lock_callbacks.<locals>.acqc                      s   ˆ   d¡ d S )NÚrelrI  r   rK  r   r   rM  ¿  s    z6TestLLVMLockCallbacks.test_lock_callbacks.<locals>.relrJ  rM  )	r   r	   Zregister_lock_callbackr4  r   rœ   Zunregister_lock_callbackr™   rj  )r   rJ  rM  r   rK  r   Útest_lock_callbacks¹  s    
z)TestLLVMLockCallbacks.test_lock_callbacksN)r5   r6   r7   rN  r   r   r   r   rI  ¸  s   rI  Ú__main__)Nrá   r   r   Úctypes.utilr   r   r   rG   rY   rO   rJ   ri   rg   Ú
contextlibr   Útempfiler   Zllvmliter   r   r   Zllvmlite.bindingr	   Zllvmlite.testsr
   Úmachinerï  r   r8   r  r  rš   rë   rw  r×   r#  rÒ  rØ  râ  r  r¤   rŽ   r’   r”   rŽ  r=  r?  r   r9   rk   r—   rÝ   Úobjectr"  rX  r]  r_  r’  rš  rŸ  r¥  r¾  rÀ  rÞ  ré  rî  rô  r  r  r  rh   r[   r"  r%  rB  rI  r5   Úmainr   r   r   r   Ú<module>   sÆ   



D(ôõýÿÿ"*I  H : C(JOGL&7	X
