B
    `                 @   s   d dl mZ d dlZd dlZd dlmZmZ d dlm	Z	m
Z
mZ d dlmZmZ dZe	dedd	d
dddddddg
d dddddddddg
gfZdd ZG dd deZG dd deZdS )    )StringION)assert_equalassert_array_almost_equal_nulp)
coo_matrix
csc_matrixrand)hb_readhb_writea%  No Title                                                                |No Key
             9             4             1             4
RUA                      100           100            10             0
(26I3)          (26I3)          (3E23.15)
1  2  2  2  2  3  3  3  3  3  3  3  3  3  3  3  3  3  3  3  3  3  3  3  3  3
3  3  3  3  3  3  3  3  3  3  3  3  3  3  3  3  3  3  3  3  3  3  3  3  3  3
3  3  3  3  3  3  3  4  4  4  6  6  6  6  6  6  6  6  6  6  6  8  9  9  9  9
9  9  9  9  9  9  9  9  9  9  9  9  9  9  9  9  9  9  9  9  9  9 11
37 71 89 18 30 45 70 19 25 52
2.971243799687726e-01  3.662366682877375e-01  4.786962174699534e-01
6.490068647991184e-01  6.617490424831662e-02  8.870370343191623e-01
4.196478590163001e-01  5.649603072111251e-01  9.934423887087086e-01
6.912334991524289e-01
)
g?gC&9kp?g)u?gMf?g"?g}b?gPٓ?gL`'?g>G?gO=?$   F   X         ,   E         3      :   =   H   I   c   c             C   s@   t | } t |}t| j|j t| j|j t| j|jd d S )Ni'  )r   r   Zindptrindicesr   data)rl r   T/home/dcms/DCMS/lib/python3.7/site-packages/scipy/io/harwell_boeing/tests/test_hb.pyassert_csc_almost_equal&   s
    r    c               @   s   e Zd Zdd ZdS )TestHBReaderc             C   s   t tt}t|t d S )N)r   r   SIMPLEr    SIMPLE_MATRIX)selfmr   r   r   test_simple/   s    zTestHBReader.test_simpleN)__name__
__module____qualname__r&   r   r   r   r   r!   .   s   r!   c               @   s   e Zd Zdd Zdd ZdS )TestHBReadWritec          	   C   sD   t jdd$}t|| |jd t|}W d Q R X t|| d S )Nzw+t)moder   )tempfileNamedTemporaryFiler	   fileseekr   r    )r$   valuer.   Zvalue_loadedr   r   r   check_save_load6   s
    
zTestHBReadWrite.check_save_loadc             C   s6   t ddd}x$dD ]}|j|dd}| | qW d S )N
   d   g?)ZcooZcscZcsrZbsrZdiaZdokZlilF)copy)r   Zasformatr1   )r$   Zrandom_matrixZmatrix_formatZmatrixr   r   r   r&   =   s    
zTestHBReadWrite.test_simpleN)r'   r(   r)   r1   r&   r   r   r   r   r*   4   s   r*   )ior   r,   numpynpZnumpy.testingr   r   Zscipy.sparser   r   r   Zscipy.ior   r	   r"   arrayr#   r    objectr!   r*   r   r   r   r   <module>   s    