U
|
ý°du ã @ sJ d dl Z d dlZd dlZd dlmZmZmZ d dlm Z
|
G dd dZ dS )é N)Ú DataFrameÚIndexÚSeriesc @ s e Zd Zdd Zdd ZdS )Ú
|
TestMatMulc C sp t tj dd¡dddgdddd gd
|
}t tj dd ¡dddd gd d gd
|
}t ||¡}t t |j|j¡dddgd d gd
|
}t ||¡ t ||j
|
¡}t t |j|j
|
j¡dddgd}t ||¡ t |j|¡}t |t sÜt|j |j¡sît|j ttd¡stt |j|j¡}t |j|¡ t |j ¡ |¡}t t |j|j¡dddgd d gd
|
}t |j|j¡ |j ¡ t¡|d< t ||¡}t t |j|j¡dddgd d gd
|
}t ||¡ | t¡}t ||¡}t t |j|j¡dddgd d gd
|
}t ||¡ t tj dd¡dd dgtdd
|
}t tj dd¡tddd dgd
|
}tjt dd t ||¡ W 5 Q R X d S )Né é ÚaÚbÚcÚpÚqÚrÚs)ÚindexÚcolumnsé ÚoneZtwo)r é é Zaligned©Úmatch) r ÚnpÚrandomZrandnÚoperatorÚmatmulÚdotÚvaluesÚtmZassert_frame_equalr r Zassert_series_equalÚ
|
isinstanceÚAssertionErrorr Úequalsr r ÚrangeZassert_almost_equalÚtolistr ÚroundZastypeÚintÚpytestÚraisesÚ
|
ValueError)Úselfr r ÚresultÚexpectedÚdfZdf2© r, ú]d:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/tests/frame/methods/test_matmul.pyÚ test_matmul sj
|
ÿ
|
ÿ ÿ ÿ ÿ
|
ÿ ""zTestMatMul.test_matmulc C st t j dd¡}t j dd¡}t|}d}tjt|d || W 5 Q R X tjt|d | ¡ | W 5 Q R X d S )Né
|
r r r z)shapes \(10, 4\) and \(5, 3\) not alignedr )r r Zrandr r% r& r' r" )r( r r r+ Úmsgr, r, r- Útest_matmul_message_shapesJ s z%TestMatMul.test_matmul_message_shapesN)Ú__name__Ú
|
__module__Ú __qualname__r. r1 r, r, r, r- r s ;r ) r Únumpyr r% Zpandasr r r Zpandas._testingZ_testingr r r, r, r, r- Ú<module> s
|
|
