o
    Eh                     @   s  d dl Z d dlZd dlZd dlZd dlmZmZmZ d dlm	Z	 d dl
Ze ZeejZejje ddZejjeedkddZejfded	efd
dZejdejejejfdd Zejdejejejfdd Zejdejejejejejje	dddfdd Zejdg dejdejejejfdd Zdd Z dd Z!edd Z"eejdg ddd Z#d d! Z$dS )"    N)can_set_localeget_locales
set_locale)ISMUSLzNeed locales)reason   z)Need multiple locales for meaningful testlc_varreturnc                 C   s
   t | S N)locale	setlocaler    r   Elib/python3.10/site-packages/pandas/tests/config/test_localization.py_get_current_locale   s   
r   c                 C   s0   t | }t|| dsJ t | }||ksJ d S )Nr   r   r   r   before_localeafter_localer   r   r   test_can_set_current_locale!      r   c                 C   s0   t | }td| dsJ t | }||ksJ d S )N r   r   r   r   r   r   test_can_set_locale_valid_set*   r   r   z$MUSL allows setting invalid LC_TIME.)Zmarksc                 C   s0   t | }td| drJ t | }||ksJ d S )Nznon-existent_localer   r   r   r   r   r   test_can_set_locale_invalid_set3   s   r   zlang,enc))it_CHzUTF-8)en_USascii)zh_CNZGB2312)it_ITz
ISO-8859-1c                 C   s0   t |}t| |ftj t |}||ksJ d S r
   )r   r   r   LC_ALL)langencr   r   r   r   r   r   test_can_set_locale_no_leakH   s   r"   c                 C   sP   dd }|   }|td| tdrJ W d    d S 1 s!w   Y  d S )Nc                   S   s   t  r
   )
ValueErrorr   r   r   r   mock_get_locale_   s   z8test_can_set_locale_invalid_get.<locals>.mock_get_locale	getlocaler   )contextsetattrr   r   )Zmonkeypatchr$   mr   r   r   test_can_set_locale_invalid_getZ   s
   
"r)   c                   C   s   t tdksJ d S )Nr   )len_all_localesr   r   r   r   test_get_locales_at_least_oneg   s   r,   c                  C   s*   t d } tt| d d ddksJ d S )Nr      )prefix)r+   r*   r   )Zfirst_localer   r   r   test_get_locales_prefixl   s   "r/   c           	   	   C   s   t  }t|j}| |f}t|s@d}tjtj|d t	| W d    n1 s+w   Y  W d    n1 s:w   Y  n+t	|}|
d\}}t|j}||f}||ks\J W d    n1 sfw   Y  t  }||kstJ d S )Nzunsupported locale setting)match.)r   codecslookupnamer   pytestZraisesr   Errorr   split)	r    r!   r   Z
new_localemsgZnormalized_localeZnew_langZnew_encr   r   r   r   test_set_localer   s(   

r9   c                  C   sH   t jd} | r| dd nd}ttjjj	j
t|j
ks"J d S )Nr   r1   zutf-8)osenvirongetr7   r2   r3   pdoptionsZdisplayencodingr4   )Zsystem_localeZsystem_encodingr   r   r   test_encoding_detected   s
   
rA   )%r2   r   r;   r5   Zpandas._config.localizationr   r   r   Zpandas.compatr   Zpandasr>   r+   r   r   Z_current_localeZmarkZskipifZ
pytestmarkr*   Z_skip_if_only_one_localeintstrr   ZparametrizeLC_CTYPELC_TIMEr   r   Zparamr   r"   r)   r,   r/   r9   rA   r   r   r   r   <module>   s`    


	
	