U
    ¬9%eô9 ã                   @   sV  d dl Z d dlZd dl mZ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Z+dZ,dZ-dZ.dZ/d Z0d!d"d#d$d%d&d'd(d)d*d+d,gZ1d!d-d.d$d/d'd(d)d*d+d,gZ2d0d1d,gZ3d2Z4d3Z5d4Z6d5Z7G d6d7„ d7eƒZ8G d8d9„ d9e8ƒZ9G d:d;„ d;e8ƒZ:G d<d=„ d=e8ƒZ;G d>d?„ d?e8ƒZ<G d@dA„ dAe=ƒZ>G dBdC„ dCe>ƒZ?G dDdE„ dEe8e?ƒZ@G dFdG„ dGe8ƒZAG dHdI„ dIe8ƒZBG dJdK„ dKe8ƒZCG dLdM„ dMe8ƒZDG dNdO„ dOe8ƒZEG dPdQ„ dQe8ƒZFG dRdS„ dSe=ƒZGG dTdU„ dUe8eGƒZHG dVdW„ dWe8eGƒZIG dXdY„ dYe8eGƒZJG dZd[„ d[e8ƒZKG d\d]„ d]e8ƒZLG d^d_„ d_e8ƒZMG d`da„ dae@ƒZNG dbdc„ dce8ƒZOe Pe
 ¡  Qdd¡de¡G dfdg„ dge8ƒƒZRG dhdi„ die8ƒZSG djdk„ dke8ƒZTG dldm„ dme8ƒZUeVdnkrRe W¡  dS )oé    N)Ú	CFUNCTYPEÚc_intÚc_int32)Ú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   úZ/var/www/html/Darija-Ai-API/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}"

    define i64 @sum(i64 %.1, i64 %.2) {{
      %.3 = add i64 %.1, %.2
      %.4 = add i64 5, %.3
      %.5 = add i64 -5, %.4
      ret i64 %.5
    }}
    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
    }}
    a  
    ; ModuleID = '<string>'
    target triple = "{triple}"
    @mul_glob = global i32 0

    declare i32 @sum(i32, i32)
    define i32 @square_sum(i32 %.1, i32 %.2) {{
      %.3 = call i32 @sum(i32 %.1, i32 %.2)
      %.4 = mul i32 %.3, %.3
      ret i32 %.4
    }}
    zÏ
    ; ModuleID = '<string>'
    target triple = "{triple}"

    declare i8* @Py_GetVersion()

    define void @getversion(i32 %.1, i32 %.2) {{
      %1 = call i8* @Py_GetVersion()
      ret void
    }}
    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 fp128 0xLF3CB1CCF26FBC178452FB4EC7F91DEAD, 0xL00000000000000000000000000000001
      ret void
    }}
    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}}]
    aƒ  
    ; ModuleID = "<string>"
    target triple = "{triple}"

    @A = external global i32

    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Æ
    ; ModuleID = '<string>'
    target triple = "{triple}"

    define void @foo(i64* %.1) {{
      ret void
    }}

    define void @bar() {{
      call void @foo(i64* null)
      ret void
    }}
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>48c1e2204809c24889d048c1c03d4831d048b90120000480001070480fafc8ay  
; ModuleID = '<lambda>'
source_filename = "<string>"
target datalayout = "e-m:e-p270:32:32-p271:32:32-p272:64:64-i64:64-f80:128-n8:16:32:64-S128"
target triple = "x86_64-pc-windows-msvc"

declare float @llvm.exp2.f32(float %casted)

define float @foo(i16 %arg) {
entry:
  %casted = sitofp i16 %arg to float
  %ret = call float @llvm.exp2.f32(float %casted)
  ret float %ret
}
c                   @   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    r1   Útargetr   r   r   Útarget_machine  s    
zBaseTest.target_machine)r-   N)	Ú__name__Ú
__module__Ú__qualname__r!   r#   Úasm_sumr,   r-   r6   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_pathrA   ÚpÚoutÚ_Zlib_patÚdepsÚlineÚpartsÚdepÚmÚallowedr   r   r   Ú
test_linux  s@    
 ÿ

   ÿ
zTestDependencies.test_linuxN)r7   r8   r9   Ú__doc__ÚunittestÚ
skipUnlessÚsysr[   r]   rI   rJ   Úgetrg   r   r   r   r   r;     s   ÿÿr;   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!   B  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   r2   Úfrom_tripler%   ÚRuntimeErrorÚstrZskipTest)r    Úer   r   r   Úcheck_riscv_targetG  s    zTestRISCVABI.check_riscv_targetc                 K   s   t j | j¡}|jf |ŽS r   )r   r2   rq   r%   r4   )r    ÚkwargZlltargetr   r   r   Úriscv_target_machineP  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fpextÚfaddr   r   r(   rs   Úverify)r    Zf64Zf32Zfntyr,   ÚfuncÚblockÚbuilderÚaÚbZarg0ÚresultÚllmodr   r   r   Úfpadd_ll_moduleT  s    


zTestRISCVABI.fpadd_ll_modulec                 C   sD   g }|  ¡ D ]2}| ¡ }| d¡s| d¡s|dks| |¡ q|S )NÚ.rx   Ú )rU   Ústripr]   Úappend)r    r)   Zasm_listrb   Zs_liner   r   r   Úbreak_up_asme  s    ÿzTestRISCVABI.break_up_asmc                 C   s8   |   ¡  |  ¡ }| jdd}|  |  | |¡¡t¡ d S )Nú+f,+d)Úfeatures)ru   r‰   rw   r"   rŽ   Úemit_assemblyÚriscv_asm_ilp32©r    rˆ   r5   r   r   r   Útest_rv32d_ilp32n  s    ÿzTestRISCVABI.test_rv32d_ilp32c                 C   s:   |   ¡  |  ¡ }| jddd}|  |  | |¡¡t¡ d S )Nr   Zilp32f©r   Zabiname)ru   r‰   rw   r"   rŽ   r‘   Úriscv_asm_ilp32fr“   r   r   r   Útest_rv32d_ilp32fu  s    ÿzTestRISCVABI.test_rv32d_ilp32fc                 C   s:   |   ¡  |  ¡ }| jddd}|  |  | |¡¡t¡ d S )Nr   Zilp32dr•   )ru   r‰   rw   r"   rŽ   r‘   Úriscv_asm_ilp32dr“   r   r   r   Útest_rv32d_ilp32d|  s    ÿzTestRISCVABI.test_rv32d_ilp32d)r7   r8   r9   rh   r%   r!   ru   rw   r‰   rŽ   r”   r—   r™   Ú__classcell__r   r   ro   r   rm   ;  s   		rm   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,   r:   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)ÚassertRaisesrr   r,   Úasm_parse_errorrs   Ú	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<>!*''#ry   i90  )r   r{   rz   ÚIntTyper|   r~   r}   r   ÚConstantrs   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_contextœ  s    

zTestMisc.test_global_contextc                 C   sH   t  dd¡ t  dd¡ t  d¡}|  |d¡ t  d¡}|  |d ¡ d S )NÚ__xyzzyéÒ  i.  Ú__foobar)r   Z
add_symbolZaddress_of_symbolr"   ÚassertIs)r    Úaddrr   r   r   Útest_dylib_symbols¡  s    

zTestMisc.test_dylib_symbolsc                 C   s"   t  ¡ }|  |t¡ |  |¡ d S r   )r   r'   ÚassertIsInstancers   Ú
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_ppc°  s    z7TestMisc.test_get_process_triple.<locals>.normalize_ppcú-r   )r   Zget_process_tripler'   rµ   rs   r¶   rV   r"   )r    rº   r%   ÚdefaultZdefault_archZtriple_archr   r   r   Útest_get_process_triple®  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Úitemsrs   r¶   ÚboolÚflattenr&   ÚassertIsNotNonerQ   rW   rX   r"   )r    r   ÚkÚvZre_termÚregexr   r   r   Útest_get_host_cpu_featuresÀ  s     

z#TestMisc.test_get_host_cpu_featuresc                 C   s"   t  ¡ }|  |t¡ |  |¡ d S r   )r   Úget_host_cpu_namerµ   rs   r¶   )r    Úcpur   r   r   Útest_get_host_cpu_name×  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©rL   Ú
check_callrk   Ú
executable©r    Úcoder   r   r   Útest_initfiniÜ  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   s4   t j\}}}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_localers   r   r   r   r"   )r    re   Ú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Ê   )rk   rÍ   rL   rÌ   )r    rÏ   ÚflagsZcmdargsr   r   r   Útest_no_accidental_warnings	  s    z$TestMisc.test_no_accidental_warningsN)r7   r8   r9   rh   rœ   r£   r¬   r®   r´   r·   r½   rÆ   rÉ   rÐ   rÑ   rÙ   rÚ   ri   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,   rs   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,   rs   Ú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,   rs   r   rå   ræ   rr   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_file1  s    zTestModuleRef.test_source_filec                 C   s2   |   ¡ }|j}|  |t¡ ||_|  ||j¡ d S r   )r,   Zdata_layoutrµ   rs   r"   râ   r   r   r   Útest_data_layout5  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   rr   rs   rŸ   r    )r    r+   r¡   r¢   r   r   r   Útest_verifyC  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   Ú	NameErrorrs   ©r    r+   rª   r   r   r   Útest_get_functionN  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Â   rQ   rW   rs   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ô   rs   )r    r+   Úgvr   r   r   Útest_get_global_variableg  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   ry   )Úvaluer   r   r   Ú<lambda>x  ó    z5TestModuleRef.test_global_variables.<locals>.<lambda>)Úkeyé   c                 S   s   g | ]
}|j ‘qS r   ry   )Ú.0Úgr   r   r   Ú
<listcomp>z  s     z7TestModuleRef.test_global_variables.<locals>.<listcomp>r-   Úglob_bZglob_fÚglob_struct)r,   Úglobal_variablesÚsortedr"   rX   )r    r+   ÚitZglobsr   r   r   Útest_global_variablest  s    
ÿz#TestModuleRef.test_global_variablesc                 C   s>   |   ¡ }|j}~t|ƒ}|  t|ƒd¡ |  |d jd¡ d S ©NrB   r   rñ   )r,   Ú	functionsÚlistr"   rX   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 )NrB   ústruct\.glob_type(\.[\d]+)?r   r÷   )
r,   Zstruct_typesr  r"   rX   rÂ   rQ   rW   r/   rs   )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   ry   ©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   ry   r  r   r   r   r  ¡  s     z6TestModuleRef.test_link_in_preserve.<locals>.<genexpr>r  rñ   c                 s   s   | ]}|j V  qd S r   ry   r  r   r   r   r  £  s     )r,   r  r  r"   r  r
  rä   rò   )r    r  Zsrc2r   r   r   Útest_link_in_preserveœ  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   rr   r  r    rs   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_bitcode¯  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   rr   r   Úparse_bitcoder    rs   rŸ   rÔ   )r    r¡   r   r   r   Útest_parse_bitcode_error¹  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_roundtripÅ  s    
z$TestModuleRef.test_bitcode_roundtripc                 C   s4   |   ¡ }| ¡ }|  ||¡ |  | ¡ | ¡ ¡ d S r   )r,   ÚcloneZassertIsNotr"   r  )r    re   Zclonedr   r   r   Útest_cloningÐ  s    zTestModuleRef.test_cloningN)r7   r8   r9   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_nameÚcfptrr   r   r   Úget_sumÜ  s    

zJITTestMixin.get_sumc              	   C   sB   |   ¡ }|  |¡&}|  |¡}|ddƒ}|  d|¡ W 5 Q R X d S ©NrC   éûÿÿÿéýÿÿÿ)r,   r1   r,  r"   )r    r+   r)  ÚcfuncÚresr   r   r   Útest_run_codeâ  s
    

zJITTestMixin.test_run_codec              	   C   sB   |   |  ¡ ¡}| ¡  | ¡  |  tj¡ | ¡  W 5 Q R X d S r   )r1   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   )r1   r,   r   rr   rå   ræ   r'  r3  r   r   r   ré   ð  s    
zJITTestMixin.test_withc                 C   s&   |   ¡ }|  |¡}| ¡  | ¡  d S r   ©r,   r1   rä   ©r    r+   r)  r   r   r   Útest_module_lifetimeú  s    
z!JITTestMixin.test_module_lifetimec                 C   s&   |   ¡ }|  |¡}| ¡  | ¡  d S r   r4  r5  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   )
r1   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   ©r1   r,   r  r8  rä   r<  r   r   r   Útest_add_module_lifetime  s
    

z%JITTestMixin.test_add_module_lifetimec                 C   s6   |   |  ¡ ¡}|  t¡}| |¡ | ¡  | ¡  d S r   r>  r<  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   )
r1   r,   r  r8  Zremove_moduler   r9  r:  r;  rä   r<  r   r   r   Útest_remove_module  s    


zJITTestMixin.test_remove_modulec                 C   s>   |   ¡ }|  |¡}|j}|  |j|¡ t|ƒ ~~t|ƒ d S r   )r,   r1   Útarget_datar²   rs   )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ÿ   é   rB   )rB   rC   rÿ   é   )rÿ   rF  )r,   r1   rB  r.   rk   Úmaxsizer"   Úget_abi_sizeÚtypeÚget_pointee_abi_sizeZget_pointee_abi_alignmentr    )	r    r+   r)  rC  Úgv_i32Zgv_i8Z	gv_structÚpointer_sizer  r   r   r   Útest_target_data_abi_enquiries3  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   ÚnotifyK  s    z5JITTestMixin.test_object_cache_notify.<locals>.notifyr   rC   r.  rB   r  )r,   r1   Úset_object_cacher"   rX   r,  r²   rµ   Úbytesr  r8  )r    rT  r+   r)  r0  Zmod2r   rR  r   Útest_object_cache_notifyH  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   rO  rP  rR  r   r   rT  g  s    z8JITTestMixin.test_object_cache_getbuffer.<locals>.notifyc                    s   ˆ   | ¡ d S r   rO  ©r+   )Ú
getbuffersr   r   Ú	getbufferj  s    z;JITTestMixin.test_object_cache_getbuffer.<locals>.getbufferr   rB   c                    s   ˆ   | ¡ ˆS r   rO  rX  )rY  Ú
sum_bufferr   r   Úgetbuffer_successful}  s    
zFJITTestMixin.test_object_cache_getbuffer.<locals>.getbuffer_successfulrC   r.  r/  )r,   r1   rU  r"   rX   r,  r²   r  )r    rT  rZ  r+   r)  r0  r\  r   )rY  rS  r[  r   Útest_object_cache_getbufferc  s4    




z(JITTestMixin.test_object_cache_getbufferN)rñ   )r7   r8   r9   rh   r,  r2  rè   ré   r6  r7  r=  r?  r@  rA  rD  rN  rW  r]  r   r   r   r   r&  ×  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()Tr0   rñ   N)r6   r,   r1   r‘   r    Zset_asm_verbosityÚassertNotEqual)r    r6   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()Tr0   r<   s   ELFNrÓ   )
r6   r,   r1   Úemit_objectrµ   rV  rk   r[   r]   r    )r    r6   r+   r)  Úcode_objectr   r   r   Útest_emit_objectž  s    
z#JITWithTMTestMixin.test_emit_objectN)r7   r8   r9   r`  rc  r   r   r   r   r^    s   r^  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 )NTr0   )r6   r   Úcreate_mcjit_compiler)r    r+   r6   r   r   r   r1   ¯  s    zTestMCJit.jit)N)r7   r8   r9   rh   r1   r   r   r   r   rd  ª  s   rd  c                   @   sª   e Zd Zedddeeeeƒd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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 )&ÚTestOrcLLJITrñ   NFc                 C   sv   t j|d|d}t  ¡ }|r$| ¡  | |jt  ¡ d¡ |¡ ||¡}	|	| }
|  	|
¡ |  
||	j¡ ||	||
ƒfS )NF)Úuse_jit_linkÚsuppress_errorsr$   )r   Úcreate_lljit_compilerÚJITLibraryBuilderZadd_current_processÚadd_irr&   r'   Úexport_symbolÚlinkr¶   r"   r/   )r    r)   r*  r6   Zadd_processZ	func_typerh  Úlljitr„   Úrtr+  r   r   r   r1   ·  s&    þÿþ ý
zTestOrcLLJIT.jitc                 C   s:   t  ¡ }t  ¡  dd¡ d¡ |d¡}|  |d d¡ d S )Nr¯   r°   r¥   )r   ri  rj  Úimport_symbolrl  rm  r"   ©r    rn  ro  r   r   r   Útest_define_symbolÊ  s    ÿ ÿzTestOrcLLJIT.test_define_symbolc              	   C   s‚   t  ¡ }|  td¡ | dd¡ W 5 Q R X t  ¡  dd¡ d¡ |d¡}|  	|d d¡ |  td¡ | dd¡ W 5 Q R X d S )NzNo such libraryr¥   r±   r¯   r°   r   zSymbols not found.*__foobar)
r   ri  ÚassertRaisesRegexrr   Úlookuprj  rp  rl  rm  r_  rq  r   r   r   Ú"test_lookup_undefined_symbol_failsÐ  s    ÿ ÿÿz/TestOrcLLJIT.test_lookup_undefined_symbol_failsc              	   C   sF   t jdkr0|  td¡ tjdd W 5 Q R X n|  tjdd¡ d S )NÚwin32zJITLink .* WindowsT)rg  )rk   r[   rs  rr   r   ri  rÂ   r   r   r   r   Útest_jit_linkÛ  s    
ÿzTestOrcLLJIT.test_jit_linkc              	   C   s8   |   ¡ \}}}| |ddƒ}|  d|¡ W 5 Q R X d S r-  )r1   r"   )r    rn  ro  r0  r1  r   r   r   r2  ã  s    
zTestOrcLLJIT.test_run_codec              	   C   sD   |   ¡ \}}}| ¡  | ¡  |  t¡ | dd¡ W 5 Q R X d S ©Nr¥   rª   )r1   rä   r   r­   rt  ©r    rn  ro  r0  r   r   r   rè   é  s
    zTestOrcLLJIT.test_closec              
   C   sj   |   ¡ \}}}| W 5 Q R X |  t¡ | W 5 Q R X W 5 Q R X |  t¡ | dd¡ W 5 Q R X d S rx  )r1   r   rr   r­   rt  ry  r   r   r   ré   ð  s    
zTestOrcLLJIT.test_withc              	   C   sÔ   |   ¡ \}}}t ¡  tjt ¡ d¡ d¡ |d¡}t	t
t
t
ƒ|d ƒddƒ}|  d|¡ |  | dd¡d d¡ |  | dd¡d d¡ |  t¡ | dd¡ W 5 Q R X |  t¡ | dd¡ W 5 Q R X d S )Nr$   r  rC   r.  iöÿÿÿrñ   r   )r1   r   rj  rk  r  r&   r'   rl  rm  r   r   r"   r_  rt  r   rr   ©r    rn  Úrt_sumZ	cfunc_sumZrt_mulr1  r   r   r   Útest_add_ir_moduleú  s"    ÿþ ýzTestOrcLLJIT.test_add_ir_modulec              	   C   sF   |   ¡ \}}}~t ¡  |  t¡ | dd¡ W 5 Q R X | ¡  d S ©Nrñ   )r1   r   r   r   rr   rt  rä   )r    rn  r{  r`   r   r   r   rA  	  s    zTestOrcLLJIT.test_remove_modulec                 C   sf   |   ¡ \}}}t ¡  tjt ¡ d¡ d¡ d¡ 	|d¡}t
tttƒ|d ƒddƒ}|  d|¡ d S )Nr$   Z
square_sumrñ   rC   r.  r   )r1   r   rj  rk  Úasm_square_sumr&   r'   rl  Zadd_jit_libraryrm  r   r   r"   rz  r   r   r   Útest_lib_depends  s    ÿþý üzTestOrcLLJIT.test_lib_dependsc                 C   s8   |   ¡ \}}}|j}|  |j|¡ t|ƒ ~t|ƒ d S r   )r1   rB  r²   rs   )r    rn  ro  r`   rC  r   r   r   rD    s    zTestOrcLLJIT.test_target_datac                 C   s†   t dƒ}t ¡ }t ¡ }| tjt ¡ d¡ dt	 
|¡¡ d¡ |d¡}|d }|  |¡ |  ttƒ|ƒƒ d¡ ~|  |jd¡ d S )Nr   r$   ÚAr¥   é   é   )r   r   ri  rj  rk  Úasm_ext_ctorsr&   r'   rp  rå   Z	addressofrl  rm  r¶   r"   r   r   r_  rû   )r    Zshared_valuern  r„   ro  r¥   r   r   r   Útest_global_ctors_dtors$  s&    ÿ þý ü
z$TestOrcLLJIT.test_global_ctors_dtorsc              	   C   s0   d}|   t|¡ | jtddd W 5 Q R X d S )Nz*Failed to materialize symbols:.*getversionÚ
getversionT)rh  )rs  rr   r1   Úasm_getversion)r    Úmsgr   r   r   Ú(test_lookup_current_process_symbol_fails5  s    z5TestOrcLLJIT.test_lookup_current_process_symbol_failsc                 C   s   |   tdd d¡ d S )Nr…  T)r1   r†  r   r   r   r   Ú"test_lookup_current_process_symbol=  s    z/TestOrcLLJIT.test_lookup_current_process_symbolc                    sh   t  ¡ ‰tjt  ¡ d‰‡‡fdd„‰ ‡ fdd„tt ¡ ƒD ƒ}|D ]}| ¡  qD|D ]}| 	¡  qVd S )Nr$   c                    s   ‡ ‡‡fdd„}|S )Nc               
      sD   g } t dƒD ]2}|  t ¡  ˆ¡ d¡ ˆdˆ › d|› ¡¡ qd S )Né2   rñ   Zsum_r`   )rÕ   r   r   rj  rk  rl  rm  )ZtrackingÚc)Úirn  Úllvm_irr   r   Údo_workE  s    ÿþ ýzDTestOrcLLJIT.test_thread_safe.<locals>.compile_many.<locals>.do_workr   )rŒ  rŽ  )rn  r  )rŒ  r   Úcompile_manyD  s    z3TestOrcLLJIT.test_thread_safe.<locals>.compile_manyc                    s   g | ]}t jˆ |ƒd ‘qS ))r5   )Ú	threadingÚThread)r   rŒ  )r  r   r   r  O  s   ÿz1TestOrcLLJIT.test_thread_safe.<locals>.<listcomp>)
r   ri  r:   r&   r'   rÕ   rI   Ú	cpu_countÚstartÚjoin)r    ZthsÚthr   )r  rn  r  r   Útest_thread_safe@  s    

ÿ
zTestOrcLLJIT.test_thread_safec                 C   sj   | j dd}|  ¡ }t |¡}t ¡  | |¡¡ d¡ |d¡}t	t
t
t
ƒ|d ƒ}|  |ddƒd¡ d S )NFr0   rñ   rC   é   é   )r6   r,   r   ri  rj  Zadd_object_imgra  rl  rm  r   r   r"   )r    r6   r+   rn  ro  rñ   r   r   r   Útest_add_object_fileV  s    
ÿþ ýz!TestOrcLLJIT.test_add_object_filec           
   	   C   s¬   | j dd}|  ¡ }| |¡}tƒ \}}zrt |d¡}| |¡ W 5 Q R X t 	|¡}t 
¡  |¡ d¡ |d¡}ttttƒ|d ƒ}	|  |	ddƒd¡ W 5 t |¡ X d S )NFr0   Úwbrñ   rC   r—  r˜  )r6   r,   ra  r   rI   ÚunlinkÚfdopenÚwriter   ri  rj  Úadd_object_filerl  rm  r   r   r"   )
r    r6   r+   Úobj_binÚ	temp_descÚ	temp_pathr  rn  ro  rñ   r   r   r   Ú$test_add_object_file_from_filesystema  s&    


ÿþ ýz1TestOrcLLJIT.test_add_object_file_from_filesystem)r7   r8   r9   r:   r   r   r1   rr  ru  rw  r2  rè   ré   r|  rA  r  rD  r„  rˆ  r‰  r–  r™  r¢  r   r   r   r   rf  µ  s*    
þ


	rf  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+d,„ Zd-d.„ Zd/d0„ Zd1d2„ Zd3d4„ Zd5S )6ÚTestValueRefc                 C   s&   |   ¡ }| d¡}|  t|ƒd¡ d S )Nr-   z@glob = global i32 0)r,   r.   r"   rs   ©r    r+   r-   r   r   r   rã   w  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ë   |  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-   r¨  r   r   r   Útest_linkageƒ  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-   rª  r   r   r   Útest_visibility  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-   r¬  r   r   r   Útest_storage_class›  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"   rs   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 r°  )r,   r.   rJ  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.   rJ  r"   r/   rÂ   rQ   rW   Úelement_type)r    r+   r-   r³  Ú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.   rJ  r"   rs   r²  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"   rs   rJ  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¶   rJ  Z
is_pointerrÂ   rQ   rW   rs   rµ  )r    r+   r¶  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 r}  )r,   rò   Úasm_sum_declarer:  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,   r:   r  r  r"   rX   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"   rX   r¶   Úis_functionr/   r   r®  Ú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ñ   rC   r   rB   ú.1Úi32ú.2)r,   rò   r¶   rÂ  r  Ú	argumentsr"   rX   Úis_argumentr/   rs   rJ  r   r®  Ú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ñ   rB   r   )r,   rò   r  rÌ  r"   rX   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—  rB   rC   rY   r   )
r,   rò   r  rÌ  rÃ  r"   rX   r¶   Zis_instructionrÅ  )r    r‚   Ú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   rY   rC   rB   rÇ  rÈ  rÉ  )r,   rò   r  rÌ  rÃ  r"   rÅ  rÄ  rX   r¶   Z
is_operandr/   rs   rJ  )r    r‚   rY   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   returnedrB   )r,   Úasm_attributesr
  r  Ú
attributesr/   r"   rÊ  )r    r+   r‚   Úattrsr   r   r   r   Útest_function_attributes*  s    





z%TestValueRef.test_function_attributesc                 C   sü   |   ¡ }|  | d¡jtjj¡ | d¡}|  |jtjj¡ t	|j
ƒd }|  |jtjj¡ t	|jƒd }|  |jtjj¡ |  t	|jƒd jtjj¡ |  t	|jƒd jtjj¡ |   t¡ d¡}t	t	|j
ƒd jƒd }|  t	|jƒd jtjj¡ d S )Nr-   rñ   r   rB   r¥   )r,   r"   r.   Z
value_kindr   Z	ValueKindZglobal_variablerò   Úfunctionr  rÌ  Zbasic_blockrÃ  ZinstructionrÄ  Zconstant_intÚasm_inlineasmZ
inline_asm)r    r+   r‚   rƒ   ÚinstZ	iasm_funcZ	iasm_instr   r   r   Útest_value_kind6  s*    ÿ
ÿÿÿzTestValueRef.test_value_kindc                 C   s   |   ¡ }|  | d¡j¡ d}|jD ]Z}|  |j¡ |jD ]B}|  |j¡ |jD ]*}|  |j¡ |jD ]}|jrf|d7 }qfqPq:q$|  	|d¡ d S )Nr-   r   rB   )
r,   r¶   r.   Úis_constantr
  rÌ  r:  rÃ  rÄ  r"   )r    r+   Zconstant_operandsr‚   rƒ   rØ  Úopr   r   r   Útest_is_constantJ  s    



zTestValueRef.test_is_constantc           	   	   C   s@  |   ¡ }| d¡}tt|jƒd jƒ}|  |d jd¡ t|d jƒ}|  |d j	¡ |  
|d j	¡ |  |d  ¡ d¡ |  t¡ |d  ¡  W 5 Q R X |   t¡}| d¡}tt|jƒd jƒ}t|d jƒd }t|d jƒd }|  | ¡ d¡ |  |jddd	¡ |jd
d}tj| dd¡ddd}|  |d	¡ d S )Nrñ   r   rB   rY   rC   r˜  T)Z
signed_intr.  FrF  Úlittle)Úsigned)r,   rò   r  rÌ  rÃ  r"   rÅ  rÄ  r¶   rÚ  r:  Úget_constant_valuer   r®  Úasm_sum3ÚintÚ
from_bytesÚto_bytes)	r    r+   r‚   rÏ  rÄ  Zposint64Znegint64Zas_u64Zas_i64r   r   r   Útest_constant_intZ  s,    


ÿzTestValueRef.test_constant_intc              	   C   s  |   t¡}| d¡}tt|jƒd jƒ}|  t|ƒd¡ |  |d jd¡ t|d j	ƒ}|  
|d j¡ |  |d  ¡ d¡ |  
|d j¡ |  |d  ¡ d¡ |   t¡}| d¡}tt|jƒd jƒd }t|j	ƒ}|  t¡ |d  ¡  W 5 Q R X |  |d jdd	d¡ d S )
Nr¥   r   rC   r€   g        rB   g…ëQ¸	@T)Zround_fp)r,   rÛ   rò   r  rÌ  rÃ  r"   rX   rÅ  rÄ  r¶   rÚ  ZassertAlmostEqualrß  Úasm_double_inaccurater   r®  )r    r+   r‚   rÏ  rÄ  rØ  r   r   r   Útest_constant_fpt  s"    




zTestValueRef.test_constant_fpc                 C   sZ   |   t¡}| d¡}tt|jƒd jƒd }t|jƒd }|  |j¡ |  	| 
¡ d¡ d S )Nrê   r   z	i64* null)r,   Úasm_null_constantrò   r  rÌ  rÃ  rÄ  r¶   rÚ  r"   rß  )r    r+   r‚   rØ  Úargr   r   r   Útest_constant_as_stringˆ  s    

z$TestValueRef.test_constant_as_stringN)r7   r8   r9   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é  r   r   r   r   r£  u  s4   	
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 )Nr¥  z(No available targets are compatible with)r   r2   rq   r   rr   r    rs   rŸ   r'   r"   r%   rä   )r    r  r¡   r%   r5   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‹   rB   r¼   )r   r2   rq   r'   r4   rÇ   )r    r5   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   r2   rq   r'   r3   rµ   r/   rs   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   r2   rq   r'   r3   rµ   Údescriptionrs   r"   rí  r   r   r   Útest_description¬  s    
zTestTarget.test_descriptionc                 C   s8   t j t  ¡ ¡}t|ƒ}|  |j|¡ |  |j|¡ d S r   )r   r2   rq   r'   rs   r    r/   rð  )r    r5   r¢   r   r   r   rã   ²  s    zTestTarget.test_strN)r7   r8   r9   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   rB  »  s    zTestTargetData.target_datac                 C   s(   |   ¡ }|  ¡ }|  | |j¡d¡ d S )NrF  )rB  r-   r"   rI  rJ  ©r    rC  r-   r   r   r   Útest_get_abi_size¾  s    z TestTargetData.test_get_abi_sizec                 C   sF   |   ¡ }|  ¡ }|  | |j¡d¡ |  d¡}|  | |j¡d¡ d S )Nrÿ   r  rG  )rB  r-   r"   rK  rJ  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   rB   rF  )rB  r-   r   r®  Zget_element_offsetrJ  rµ  r"   )r    rC  r-   Zstruct_typer   r   r   Útest_get_struct_element_offsetÌ  s    
z-TestTargetData.test_get_struct_element_offsetN)r7   r8   r9   rB  rô  rõ  rö  r   r   r   r   rò  ¹  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 )NFr0   )r6   r   Úcreate_module_pass_managerÚadd_analysis_passes)r    ÚtmÚpmr   r   r   Útest_add_analysis_passesÚ  s    z*TestTargetMachine.test_add_analysis_passesc                 C   sN   | j dd}|j}|  ¡ }| d¡}tjdk r2dnd}|  | |j¡|¡ d S )NFr0   r-   rE  rÿ   rF  )	r6   rB  r,   r.   rk   rH  r"   rI  rJ  )r    rú  rC  r+   rL  rM  r   r   r   Útest_target_data_from_tmß  s    
z*TestTargetMachine.test_target_data_from_tmN)r7   r8   r9   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   Úpmbë  s    zTestPassManagerBuilder.pmbc                 C   s   t  ¡ }d|_d|_d S )NrC   r—  )r   Úcreate_pass_manager_builderÚinlining_thresholdÚ	opt_level©r    rÿ  r   r   r   Útest_old_apiî  s    z#TestPassManagerBuilder.test_old_apic                 C   s   |   ¡ }| ¡  | ¡  d S r   )rÿ  rä   r  r   r   r   rè   ô  s    z!TestPassManagerBuilder.test_closec                 C   s<   |   ¡ }|  |jt¡ tdƒD ]}||_|  |j|¡ qd S r¾  )rÿ  rµ   r  rá  rÕ   r"   ©r    rÿ  rŒ  r   r   r   Útest_opt_levelù  s
    z%TestPassManagerBuilder.test_opt_levelc                 C   s<   |   ¡ }|  |jt¡ tdƒD ]}||_|  |j|¡ qd S r¾  )rÿ  rµ   Z
size_levelrá  rÕ   r"   r  r   r   r   Útest_size_level   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)r7   r8   r9   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d„ Zdd„ ZdS )ÚPassManagerTestMixinc                 C   s   t  ¡ }d|_d|_|S )NrC   i,  )r   r   r  r  r  r   r   r   rÿ  5  s    zPassManagerTestMixin.pmbc                 C   s   |   ¡ }| ¡  | ¡  d S r   )rû  rä   ©r    rû  r   r   r   rè   ;  s    zPassManagerTestMixin.test_closeN)r7   r8   r9   rÿ  rè   r   r   r   r   r  3  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û  C  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,   rs   ÚrunrU   Ú	enumeraterŒ   r]   rr   Úpopr    ÚassertNotIn)r    rû  r+   Úorig_asmÚopt_asmZopt_asm_splitÚidxÚlÚtoksrî  Z
othertokenr   r   r   Útest_runF  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_inlined  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_inlineo  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.*r2  r%  r&  r'  r+  r   r   r   Ú&test_run_with_remarks_inline_filter_in…  s    


z<TestModulePassManager.test_run_with_remarks_inline_filter_inN)	r7   r8   r9   rû  r#  r.  r0  r3  r4  r   r   r   r   r  A  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ä   rs   r   r  r6  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*  r6  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 )Nr7  r1  r2  r‹   )r,   r9  rò   rû  r:  rÿ  r  rä   r   r*  r6  r¶   r"   r;  r   r   r   Ú test_run_with_remarks_filter_out¹  s    



z8TestFunctionPassManager.test_run_with_remarks_filter_outc                 C   s„   |   t¡}| d¡}|  |¡}| ¡  |  ¡  |¡ | ¡  | ¡  |j	|dd\}}| 
¡  |  |¡ |  d|¡ |  d|¡ d S )Nr7  r2  r%  r8  r;  r   r   r   Útest_run_with_remarks_filter_inÇ  s    



z7TestFunctionPassManager.test_run_with_remarks_filter_in)N)	r7   r8   r9   rû  rÐ   r#  r=  r>  r?  r   r   r   r   r5    s   
r5  c                   @   s:   e Zd Zdd„ Zdd„ Ze e ¡  	d¡d¡dd„ ƒZ
d	S )
Ú
TestPassesc                 C   s   t  ¡ S r   r  r   r   r   r   rû  Ù  s    zTestPasses.pmc                 C   s4  |   ¡ }| d¡ | ¡  | ¡  | ¡  | d¡ | ¡  | ¡  | ¡  | 	¡  | 
¡  | ¡  | ¡  | ¡  | ¡  | ¡  | ¡  | ¡  | ¡  | ¡  | ¡  | ¡  | ¡  | ¡  | d¡ | ¡  | ¡  | ¡  | ¡  | ¡  | d¡ | ¡  |  ¡  | !¡  | "¡  | #¡  | $¡  | %¡  | &¡  | '¡  | (¡  | )¡  | *¡  | +¡  | ,¡  | -¡  | .¡  | /¡  | 0¡  | 1¡  | 2¡  | 3¡  | 4¡  | 5¡  | 6¡  | 7¡  | 8¡  | 9¡  | :¡  | ;¡  | <¡  | =¡  | >¡  | ?¡  | @¡  | A¡  | B¡  | C¡  | D¡  d S )Nr‹   éá   é*   é   )Erû  Úadd_target_library_infoZ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_passÚ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_populateÜ  sŠ    



zTestPasses.test_populateÚx86zx86 onlyc                 C   sP   dd„ }|dd}|   dt|ƒ¡ |dd}|  dt|ƒ¡ |   dt|ƒ¡ dS )	zŠTest a specific situation that demonstrate TLI is affecting
        optimization. See https://github.com/numba/numba/issues/8898.
        c                 S   sX   t  t¡}t j |j¡}| ¡ }t  ¡ }| |¡ | rB| 	|j¡ | 
¡  | |¡ |S r   )r   r(   Úasm_tli_exp2r2   rq   r%   r4   ZModulePassManagerrù  rD  rE  r  )Úuse_tlir+   r5   rú  rû  r   r   r   r  (  s    


z9TestPasses.test_target_library_info_behavior.<locals>.runT)rI  zcall float @llvm.exp2.f32Fzcall float @ldexpfN)r    rs   r  )r    r  r+   r   r   r   Ú!test_target_library_info_behavior#  s    

z,TestPasses.test_target_library_info_behaviorN)r7   r8   r9   rû  rF  ri   rj   r[   Úmachiner]   rJ  r   r   r   r   r@  ×  s   G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   rr   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 )Nre   )r   r   rM  )r    rE   r   r   r   Ú	test_libmD  s    zTestDylib.test_libmN)	r7   r8   r9   rN  ri   rj   r[   ÚsystemrP  r   r   r   r   rL  >  s
   ÿrL  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   r{   rz   r¦   r|   r~   r}   r   rY   r   )r    re   r©   rª   r«   ÚxÚyÚzr   r   r   Úbuild_ir_moduleL  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)rV  Z
get_globalr   Úget_function_cfgr    r  )r    r+   r¥   Údot_showing_instÚdot_without_instrØ  r   r   r   Útest_get_function_cfg_on_irV  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ñ   TrW  FZdigraphzCFG for 'sum' functionz%.3 = add i32 %.1, %.2)r,   rò   r   rX  r    r  )r    r¼  rY  rZ  ÚprefixÚfnamerØ  r   r   r   Útest_function_cfg_on_llvm_value_  s    z,TestAnalysis.test_function_cfg_on_llvm_valueN)r7   r8   r9   rV  r[  r^  r   r   r   r   rR  K  s   
	rR  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   r{   rs   r   r(   )r    r+   r)   r   r   r   Úcheck_parsingr  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¥   rB   T)Úpacked)r`  r   ZLiteralStructTyper¦   ÚGlobalVariabler§   Zinitializer)r    r+   Útyprù   r   r   r   Útest_literal_struct|  s    

ÿz#TestTypeParsing.test_literal_structN)r7   r8   r9   r   r`  rd  r   r   r   r   r_  q  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 )Nr€  rÓ   r¥   r  r‚  )r,   Úasm_global_ctorsr1   r'  Zrun_static_constructorsZget_global_value_addressrå   ZPOINTERr   Úcastr"   Úcontentsrû   r(  r   Zrun_static_destructors)r    r+   r)  Zptr_addrZptr_tZptrZfoo_addrr¥   r   r   r   r„  Ž  s    



z.TestGlobalConstructors.test_global_ctors_dtorsN)r7   r8   r9   r„  r   r   r   r   re    s   re  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   r{   r¦   rb  r¨  rs   r    r  r,   )r    r¨  Ú	has_undefr+   rc  rù   r)   r   r   r   Úcheck_global_variable_linkage¨  s    
z1TestGlobalVariables.check_global_variable_linkagec                 C   s   |   d¡ d S )Nr¦  ©rk  r   r   r   r   Útest_internal_linkage¼  s    z)TestGlobalVariables.test_internal_linkagec                 C   s   |   d¡ d S )NÚcommonrl  r   r   r   r   Útest_common_linkage¿  s    z'TestGlobalVariables.test_common_linkagec                 C   s   | j ddd d S )Nr§  F©rj  rl  r   r   r   r   Útest_external_linkageÂ  s    z)TestGlobalVariables.test_external_linkagec                 C   s   |   d¡ d S )NZavailable_externallyrl  r   r   r   r   Ú!test_available_externally_linkageÅ  s    z5TestGlobalVariables.test_available_externally_linkagec                 C   s   |   d¡ d S )NÚprivaterl  r   r   r   r   Útest_private_linkageÈ  s    z(TestGlobalVariables.test_private_linkagec                 C   s   |   d¡ d S )NZlinkoncerl  r   r   r   r   Útest_linkonce_linkageË  s    z)TestGlobalVariables.test_linkonce_linkagec                 C   s   |   d¡ d S )NZweakrl  r   r   r   r   Útest_weak_linkageÎ  s    z%TestGlobalVariables.test_weak_linkagec                 C   s   |   d¡ d S )NZ	appendingrl  r   r   r   r   Útest_appending_linkageÑ  s    z*TestGlobalVariables.test_appending_linkagec                 C   s   | j ddd d S )NZextern_weakFrp  rl  r   r   r   r   Útest_extern_weak_linkageÔ  s    z,TestGlobalVariables.test_extern_weak_linkagec                 C   s   |   d¡ d S )NZlinkonce_odrrl  r   r   r   r   Útest_linkonce_odr_linkage×  s    z-TestGlobalVariables.test_linkonce_odr_linkagec                 C   s   |   d¡ d S )NZweak_odrrl  r   r   r   r   Útest_weak_odr_linkageÚ  s    z)TestGlobalVariables.test_weak_odr_linkageN)T)r7   r8   r9   rk  rm  ro  rq  rr  rt  ru  rv  rw  rx  ry  rz  r   r   r   r   ri  §  s   
ri  rG  z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)   Fr0   Znop)r   Zinitialize_native_asmparserr,   r×  r6   r‘   r    )r    re   rú  r)   r   r   r   Útest_inlineasmà  s
    
zTestInlineAsm.test_inlineasmN)r7   r8   r9   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 )NFr0   éÿÿÿÿTr   )r6   r,   ra  r   ÚObjectFileRefÚ	from_dataÚsectionsÚis_textrÂ   r/   r¶   ÚsizerX   ÚdataÚaddress)r    r6   r+   rŸ  ÚobjZhas_textZlast_addressr¢   r   r   r   Útest_object_file÷  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 )NFr0   Ú	sum_twicerC   r—  rÓ   )r6   r,   ra  r   r  r€  re  Úmod_asmrž  r   r   r(  r"   )r    r6   r+   rŸ  r†  r1   rˆ  r   r   r   r™  	  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 )NFr0   rš  rˆ  rC   r—  rÓ   )r6   r,   ra  r   rI   r›  rä   rœ  r  Úflushr   re  r‰  rž  r   r   r(  r"   )	r    r6   r+   rŸ  r   r¡  r  r1   rˆ  r   r   r   r¢  	  s&    



ÿ
ÿz3TestObjectFile.test_add_object_file_from_filesystemc                 C   sX   t  t¡}tj |¡}| ¡ D ]4}| ¡ r|  t	| 
¡ ƒd¡ |  | 
¡  ¡ t¡ qd S )Né   )rV  ÚfromhexÚissue_632_elfr   r  r€  r  r‚  r"   rX   r„  ÚhexÚissue_632_text)r    Úelfr†  r¢   r   r   r   Útest_get_section_content5	  s    
z'TestObjectFile.test_get_section_contentN)r7   r8   r9   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 )ÚTestTimePassesc              	   C   s   t  ¡ }t  ¡ }d|_| |¡ z8t  d¡ | |  ¡ ¡ | |  ¡ ¡ | |  ¡ ¡ W 5 t  ¡ }t  d¡ X |  	|t
¡ |  | d¡d¡ d S )Nr—  FTzPass execution timing reportrB   )r   rø  r   r  r  Úreport_and_reset_timingsZset_time_passesr  r,   rµ   rs   r"   Úcount)r    Úmprÿ  Úreportr   r   r   Útest_reportingA	  s    

zTestTimePasses.test_reportingc                 C   s   |   t ¡ ¡ d S r   )r:  r   r“  r   r   r   r   Útest_empty_reportT	  s    z TestTimePasses.test_empty_reportN)r7   r8   r9   r—  r˜  r   r   r   r   r’  @	  s   r’  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ÚacqrO  r   ©Úeventsr   r   rš  ]	  s    z6TestLLVMLockCallbacks.test_lock_callbacks.<locals>.acqc                      s   ˆ   d¡ d S )NÚrelrO  r   r›  r   r   r  `	  s    z6TestLLVMLockCallbacks.test_lock_callbacks.<locals>.relrš  r  )	r   r
   Zregister_lock_callbackr:  rø  r    Zunregister_lock_callbackr   r®  )r    rš  r  r   r›  r   Útest_lock_callbacksZ	  s    
z)TestLLVMLockCallbacks.test_lock_callbacksN)r7   r8   r9   rž  r   r   r   r   r™  Y	  s   r™  Ú__main__)Xrå   r  r   r   r   Úctypes.utilr   r   r   rI   r[   rQ   rL   rk   ri   Ú
contextlibr   Útempfiler   Zllvmliter   r	   r   Zllvmlite.bindingr
   Zllvmlite.testsr   rK  rM  r   r:   r  rà  r  r~  r†  rž   rï   r»  rå  rÛ   r×  r)  r/  r9  rf  rƒ  r¨   rç  r’   r–   r˜   rÒ  r  r  rH  r   r;   rm   r›   rá   Úobjectr&  r^  rd  rf  r£  rê  rò  r÷  rþ  r  r  r5  r@  rL  rR  r_  re  ri  rj   r]   r{  r}  r’  r™  r7   Úmainr   r   r   r   Ú<module>   sÜ   




D(ôõýÿÿ"*I  H : A  (JOGg&7	X
