test_capabilities.cpython-313-pytest-8.3.5.pyc•7.51 kB
�
u�h/ �
�` � S SK rS SKJs Jr S SKrS SKJr S SK J
r
J r S SK r S SK
r
\ R R S \
R R! \
R R# \
R R% \5 SS5 5 5 \R( S 5 rS rS rS rS
rg) � N)�Path)� hdf5_handler�
slurm_handlerz..�srcc �| � U S- nUR 5 US- R 5 US- R 5 U$ )N�hdf5z
data1.hdf5z
data2.hdf5)�mkdir�touch)�tmp_path�hdf5_dirs �A/Users/jafaralzoubi/Desktop/mcp-server/tests/test_capabilities.py�
mock_hdf5_dirr s= � ��&� �H� �N�N��
� � �#�#�%�
� � �#�#�%� �O� c �b � U R [ l [ R " SS05 nUS nSo2U:H oD( do [ R
" SU4SX#45 [ R " U5 [ R " U5 S.- nSS U0- n[ [ R " U5 5 eS =n=pCUS
S n[ U5 nS o�U:H o�( d� [ R
" SU 4S
X845 S[ R " 5 ; d [ R " [ 5 ( a [ R " [ 5 OS[ R " U5 [ R " U5 [ R " U5 S.- n
SSU
0- n [ [ R " U 5 5 eS =n=n=p�g )N� path_patternzhdf5/*�status�success��==�z%(py1)s == %(py4)s��py1�py4�assert %(py6)s�py6�data�files� �z0%(py4)s
{%(py4)s = %(py0)s(%(py2)s)
} == %(py7)s�len��py0�py2r �py7�assert %(py9)s�py9)
�parentr �MOCK_DATA_ROOT�handle�
@pytest_ar�_call_reprcompare� _saferepr�AssertionError�_format_explanationr � @py_builtins�locals�_should_repr_global_name) r �result� @py_assert0� @py_assert3� @py_assert2� @py_format5� @py_format7� @py_assert1� @py_assert6� @py_assert5� @py_format8� @py_format10s r
�test_hdf5_handler_successr= s � �"/�"6�"6�L� �
�
�
�.�(�!;�
<�F� �(� �(�y�(�y� (�(�(�(�(� �(�(�(� �(�(�(�y�(�(�(�(�(�(�(��f�~�g�&�,�3�&� '�,�1�,�1� ,�,�,�,�,� '�,�,�,�,�,�,�3�,�,�,�,�3�,�,�,�&�,�,�,� '�,�,�,�1�,�,�,�,�,�,�,r c �F � [ R " SS05 n U S nSo!U:H o3( do [ R " SU4SX45 [ R " U5 [ R " U5 S.- nSS U0- n[ [ R " U5 5 eS =n=p2S
oS o!U; o3( do [ R " S U4S
X45 [ R " U5 [ R " U5 S.- nSS U0- n[ [ R " U5 5 eS =n=p2g )Nr z/invalid/path/*r �errorr r r r r zInvalid base directory�message��in�z%(py1)s in %(py4)s)r r) r* r+ r, r- r. �r2 r3 r4 r5 r6 r7 s r
� test_hdf5_handler_invalid_pathrE s� � �
�
�
�.�2C�!D�
E�F� �(� �&�w�&�w� &�&�&�&�&� �&�&�&� �&�&�&�w�&�&�&�&�&�&�&� #�8�i�'8�8�'8� 8�8�8�8�8� #�8�8�8� #�8�8�8�'8�8�8�8�8�8�8�8r c �� � SSS.n [ R " U 5 nUS nSo2U:H oD( do [ R " SU4SX#45 [ R " U5 [ R " U5 S.- nS S
U0- n[ [ R " U5 5 eS =n=pCUS S n[ U5 nS
o�U:H o�( d� [ R " SU 4SX845 S[ R " 5 ; d [ R " [ 5 ( a [ R " [ 5 OS[ R " U5 [ R " U5 [ R " U5 S.- n
SSU
0- n [ [ R " U 5 5 eS =n=n=p�[ R owR o1S S o�" U 5 n S o�U
Lo�( Gd [ R " SU4SX�45 S[ R " 5 ; d [ R " [ 5 ( a [ R " [ 5 OS[ R " U5 [ R " U5 [ R " U 5 [ R " U 5 [ R " U
5 S.- nSSU0- n[ [ R " U5 5 eS =n=n=n =n =p�g )Nz /jobs/run.sh� )� script_path�coresr r r r r r r r �job_id�
r r r! r% r& )�is not)zk%(py8)s
{%(py8)s = %(py4)s
{%(py4)s = %(py2)s
{%(py2)s = %(py0)s.jobs_db
}.get
}(%(py6)s)
} is not %(py11)sr )r" r# r r �py8�py11zassert %(py13)s�py13)
r r) r* r+ r, r- r. r r/ r0 r1 �jobs_db�get)�paramsr2 r3 r4 r5 r6 r7 r8 r9 r: r; r< � @py_assert7� @py_assert10� @py_assert9� @py_format12� @py_format14s r
�test_slurm_handler_successrX s� � � +�a�
8�F�
�
!�
!�&�
)�F� �(� �(�y�(�y� (�(�(�(�(� �(�(�(� �(�(�(�y�(�(�(�(�(�(�(��f�~�h�'�.�3�'� (�.�B�.�B� .�.�.�.�.� (�.�.�.�.�.�.�3�.�.�.�.�3�.�.�.�'�.�.�.� (�.�.�.�B�.�.�.�.�.�.�.� � � �J� $� $�J�F�^�H�%=�J� $�%=� >�J�d�J�d� J�J�J�J�J� >�J�J�J�J�J�J�=�J�J�J�J�=�J�J�J� �J�J�J� $�J�J�J�%=�J�J�J� >�J�J�J�d�J�J�J�J�J�J�J�Jr c �F � [ R " SS05 n U S nSo!U:H o3( do [ R " SU4SX45 [ R " U5 [ R " U5 S.- nSS U0- n[ [ R " U5 5 eS =n=p2S
oS o!U; o3( do [ R " S U4S
X45 [ R " U5 [ R " U5 S.- nSS U0- n[ [ R " U5 5 eS =n=p2g )NrI � r r? r r r r r rH r@ rA rC )r r) r* r+ r, r- r. rD s r
�!test_slurm_handler_missing_scriptr[ ! s� � �
�
!�
!�7�A�,�
/�F� �(� �&�w�&�w� &�&�&�&�&� �&�&�&� �&�&�&�w�&�&�&�&�&�&�&� �-�9� -�-� -� -�-�-�-�-�=�-�-�-�=�-�-�-� -�-�-�-�-�-�-�-r )�builtinsr/ �_pytest.assertion.rewrite� assertion�rewriter* �pytest�pathlibr �src.capabilitiesr r �sys�os�path�insert�abspath�join�dirname�__file__�fixturer r= rE rX r[ � r r
�<module>rm s� �� � �
� � 8�
� � ������2�7�7�?�?�2�7�7�<�<�������0I�4�QV�#W�X� Y�� � �� ��-� 9�
K�.r