o
    5ήc                     @   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g dg d	gfZd
d ZG dd dZG dd d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   )
r      :   =   r   H   r   I   c   r   c                 C   s@   t | } t |}t| j|j t| j|j t| j|jd d S )Ni'  )r   r   indptrindicesr   data)rl r   M/tmp/pip-target-vg8gfxp4/lib/python/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   sX   t jdd}t|| |jd t|}W d    n1 s w   Y  t|| d S )Nzw+t)moder   )tempfileNamedTemporaryFiler	   fileseekr   r!   )r%   valuer/   value_loadedr   r   r    check_save_load6   s   

zTestHBReadWrite.check_save_loadc                 C   s2   t ddd}dD ]}|j|dd}| | qd S )N
   d   g?)coocsccsrbsrdiadoklilF)copy)r   asformatr3   )r%   random_matrixmatrix_formatmatrixr   r   r    r'   =   s
   zTestHBReadWrite.test_simpleN)r(   r)   r*   r3   r'   r   r   r   r    r+   4   s    r+   )ior   r-   numpynpnumpy.testingr   r   scipy.sparser   r   r   scipy.ior   r	   r#   arrayr$   r!   r"   r+   r   r   r   r    <module>   s"    
