+
�n�iV c �� � R t ^ RIt^ RIHu Ht ^ RIt^ RI t ^ RI
H t Rt RR R llt
RR R lltR R ltRR
R llt] P" R R
l4 t] P" R R l4 tR# )z7Shared fixtures and helpers for orion_sdr pytest tests.N)�NDArrayg p�@c �~ � V ^8� d QhR\ R\ R\ R\ R\ \ P , /# �� �fs�f_hz�n�amp�return)�float�intr �np�float32)�formats "�o/Users/glenn/Library/Mobile Documents/com~apple~CloudDocs/work/rust/projects/orion-sdr/python/tests/conftest.py� __annotate__r s? � � E� E�%� E�u� E�� E�5� E�7�2�:�:�CV� E� c � � \ P ! V\ P R7 V , pV\ P ! R\ P , V, V, 4 , P \ P 4 # )z Generate a real sinusoidal tone.��dtype� @)r �aranger
�sin�pi�astype�r r r r �ts &&&& r � real_toner sP � �
� � �!�2�:�:�&��+�A� �"�&�&��r�u�u��t�+�a�/�0� 0� 8� 8���� D�Dr c �~ � V ^8� d QhR\ R\ R\ R\ R\ \ P , /# r )r
r r r � complex64)r s "r r r s? � � L� L�U� L�%� L�C� L�e� L�g�b�l�l�F[� Lr c � � \ P ! V\ P R7 V , pV\ P ! R\ P , V, V, 4 , P \ P 4 # )z2Generate a complex exponential tone (baseband IQ).r y @)r r r
�expr r r r s &&&& r � complex_toner! sQ � �
� � �!�2�:�:�&��+�A� �"�&�&��B�E�E�)�D�0�1�4�5� 5� =� =�b�l�l� K�Kr c �r � V ^8� d QhR\ \ P , R\ R\ R\ /# )r �xr r r )r r r
r
)r s "r r r s1 � �
*�
*�g�b�j�j�!�
*�u�
*�E�
*�e�
*r c � a aaa� \ S 4 o\ P ! S\ P R7 oR VVVV 3R llpV! V4 pV! VR, 4 R, pR\ P ! WE, 4 , # )zCSingle-bin DFT SNR estimate: power at f_hz vs power at f_hz * 0.73.r c �0 � V ^8� d QhR\ R\ /# )r �fr )r
)r s "r r � snr_db.<locals>.__annotate__ s � � � �U� �u� r c �b <� R\ P , V , S, S, p\ P ! V4 R\ P ! V4 , , p\ \ P
! \ P ! VS4 4 ^, 4 SS, , pV# )r y �?g �)r r �cosr r
�abs�dot)r& �w�phasor�cr �kr r# s & ����r � dft_power�snr_db.<locals>.dft_power sn �� � �2�5�5�L�1� �r� !�A� %�������R�"�&�&��)�^�+�� �"�&�&������*�+�q�0� 1�Q��U� ;���r g\���(\�?g#B� ���;g $@)�lenr r �float64�log10)r# r r r0 �p_sig�p_offr/ r s ff& @@r �snr_dbr7 s^ �� � �A��A�
� � �!�2�:�:�&�A�� �
�d�O�E� �d�T�k� "�U� *�E� �"�(�(�5�=�)� )�)r c �< � V ^8� d QhR\ R\ R\ /# )r r# �fractionr )r r
)r s "r r r ' s! � � .� .�G� .�u� .�� .r c �L � V \ \ V 4 RV,
, 4 R # )zFDrop the leading transient, return the trailing fraction of the array.� �?N)r r2 )r# r9 s &&r �tailr< ' s" � � �S��Q��3��>�*�
+�
,� -�-r c �N � V ^8� d QhR\ \ P , /# �r r )r r r )r s "r r r - s � � .� .�G�B�L�L�)� .r c �N � \ P ! R \ P R7 # �i r )r �zerosr � r r �
silence_iqrC , s � �
�8�8�D�� � � -�-r c �N � V ^8� d QhR\ \ P , /# r> )r r r
)r s "r r r 2 s � � ,� ,�w�r�z�z�*� ,r c �N � \ P ! R \ P R7 # r@ )r rA r
rB r r �
silence_audiorF 1 s � �
�8�8�D��
�
� +�+r )r; )g �?)�__doc__�builtins� @py_builtins�_pytest.assertion.rewrite� assertion�rewrite�
@pytest_ar�numpyr �pytest� numpy.typingr �FSr r! r7 r< �fixturerC rF rB r r �<module>rS sc �� =� � � �
� � ��E� L�
*� .�
� � �.� �.� � � �,� �,r