
    _-Ph                     ^    d Z ddlZddlmZ ddlmZmZmZ ddl	m
Z
mZ  G d de
          ZdS )zTest BSR functions.    N)
bsr_matrix)
bsr_getrowbsr_row_setscalarbsr_row_setvector)TestCaseassert_equalc                        e Zd Zd Zd Zd ZdS )TestBSRUtilsc                    t          j        g d          }t          j        g d          }t          j        g d                              d                              ddd          }t	          |||fd          }t          |d          \  }}t          |t          j        d	gd	gg                     t          |t          j        dd
g                     d S )Nr            r   r   r   r      r   r   r   r         r   r   r   r   r   r   shaper   r   )nparrayrepeatreshaper   r   r   )selfindptrindicesdataBris          _/var/www/html/test/jupyter/venv/lib/python3.11/site-packages/pyamg/util/tests/test_bsr_utils.pytest_bsr_getrowzTestBSRUtils.test_bsr_getrow
   s    ,,,''(---..x***++22155==aAFFgv.f===!Q1Q1#s,,---Q!Q(()))))    c           
         t          j        g d          }t          j        g d          }t          j        g d                              d                              ddd          }t          j        g d          }t          j        g d          }t          j        ddgddggddgddggddgddggddgd	d	ggd
d
gd	d	ggddgd	d	ggg          }t	          |||fd          }t	          |||fd          }t          |d
d	           t          j        ||z
  j                  }	t          |	j	        d         d           d S )Nr   r   r   r   r   r   r   r      r   r   r   r   )
r   r   r   r   r   r   ravelr   r   r   
r   r   r   r   indptr2indices2data2B2r    diffs
             r#   test_bsr_row_setscalarz#TestBSRUtils.test_bsr_row_setscalar   sv   ,,,''(---..x***++22155==aAFF(<<<((8...//Aq6Aq6*q6Aq6*q6Aq6*q6B8,q6B8,q6B8,. / / '2&AAAgv.f===!Q###xR&&TZ]A&&&&&r%   c           
         t          j        g d          }t          j        g d          }t          j        g d                              d                              ddd          }t          j        g d          }t          j        g d          }t          j        ddgddggddgddggddgddggddgd	d
ggddgddggddgddggg          }t	          |||fd          }t	          |||fd          }t          |dt          j        g d                     t          j        ||z
  j                  }	t          |	j	        d         d           d S )Nr   r   r   r   r   r   r   r      r'   r   !   ,   7   B   r   r   )r1   r'   r2   r3   r4   r5   r   )
r   r   r   r   r   r   r(   r   r   r   r)   s
             r#   test_bsr_row_setvectorz#TestBSRUtils.test_bsr_row_setvector'   s   ,,,''(---..x***++22155==aAFF(<<<((8...//Aq6Aq6*q6Aq6*q6Aq6*q6B8,q6B8,q6B8,. / / '2&AAAgv.f===!Q)A)A)A B BCCCxR&&TZ]A&&&&&r%   N)__name__
__module____qualname__r$   r/   r6    r%   r#   r
   r
   	   sA        * * *' ' '(' ' ' ' 'r%   r
   )__doc__numpyr   scipy.sparser   pyamg.util.bsr_utilsr   r   r   numpy.testingr   r   r
   r:   r%   r#   <module>r@      s          # # # # # # Q Q Q Q Q Q Q Q Q Q 0 0 0 0 0 0 0 00' 0' 0' 0' 0'8 0' 0' 0' 0' 0'r%   