← Back
Editing: test_qmc.cpython-311.pyc
� d�c�� � � � d dl Z d dlmZ d dlmZmZ d dlZd dlZd dl m Z mZmZ d dl mZ d dlmZ d dlmZ d dlmZ d d lmZmZmZmZmZmZmZmZ G d � d� � Z G d� d � � Z G d� dej � � Zd� Z d� Z!d� Z"d� Z# G d� d� � Z$ G d� de$� � Z% G d� de$� � Z& G d� de$� � Z' G d� de$� � Z( G d� d� � Z) G d � d!� � Z* G d"� d#� � Z+ G d$� d%� � Z,d&� Z-dS )'� N)�Counter)�combinations�product)�assert_allclose�assert_equal�assert_array_equal)�distance)�shapiro)�_test_find_index)�qmc)�van_der_corput�n_primes�primes_from_2_to�update_discrepancy� QMCEngine�_l1_norm�_perturb_discrepancy�&_lloyd_centroidal_voronoi_tessellationc �P � e Zd Zd� Zd� Zd� Zd� Zd� Zd� Zd� Z d� Z d � Zd � Zd� Z dS ) � TestUtilsc �( � dgdgdgg}dgdgdgg}t j |dd�� � }t ||� � ddgddgddgg}t j ddgddgg� � }ddgddgdd gg}t j ||d |d �� � }t ||� � t j ||d |d d �� � }t ||� � g d�g d �g d�g}dg d�}}g d�g d�g d�g}t j |||�� � }t ||� � d S )Nr � � �?���� � ��l_bounds�u_bounds� � @T�r r �reverse�r r r )r r r )r r r )r r � )r% r! � �?)r �scaler �np�array)�self�space�out�scaled_space�bounds�scaled_back_spacer r s �</usr/lib/python3/dist-packages/scipy/stats/tests/test_qmc.py� test_scalezTestUtils.test_scale so � ���q�c�C�5�!���t�a�S�1�#����y���a�@�@�@����c�*�*�*� �Q��!�Q��#�s��,����B��7�Q��F�+�,�,���A�w��A���C��)���y�����V�A�Y�O�O�O����c�*�*�*��I�l�V�A�Y�/5�a�y�$�H� H� H���)�5�1�1�1� ���I�I�I����7��� � � �(���y�y�)�)�)�]�]�]�3���y���H�M�M�M����c�*�*�*�*�*� c �P � t j � d� � }|� d� � }|� d� � dz }|� d� � dz }t j |||d�� � }t j |||d�� � }t ||� � d S )Nl Ea�T4�G��*�w�v� �� � r6 F)r# T)r( �random�default_rngr r'