o
    }oi                     @   s   d dl mZ d dlZd dlmZmZmZ d dlmZ ej	
dedddgdd	 Zej	
d
edddgdd Zej	
d
edddgdd Zej	
d
edddgdd ZdS )    )PathN)ckpt_to_context_subdirckpt_to_diridempotent_path_append)ckpt_to_weights_subdirbase_dirz	test/pathzmsc://default/tmp/test/pathc                 C   s$   d}t | |}t|dsJ d S )N
new_suffixztest/path/new_suffix)r   strendswith)r   suffixresult r   S/home/ubuntu/.local/lib/python3.10/site-packages/tests/lightning/test_ckpt_utils.py*test_idempotent_path_append_path_no_suffix   s   
r   filepathtest/checkpointsz"msc://default/tmp/test/checkpointsc                 C      t | }t|dsJ d S )Nztest/checkpoints/context)r   r	   r
   r   r   r   r   r   test_ckpt_to_context_subdir      r   c                 C   r   )Nr   )r   r	   r
   r   r   r   r   test_ckpt_to_dir&   r   r   c                 C   s"   t | dd}t|dsJ d S )NT)	is_savingztest/checkpoints/weights)r   r	   r
   r   r   r   r   test_ckpt_to_weights_subdir.   s   r   )pathlibr   pytestnemo.lightning.ckpt_utilsr   r   r   nemo.lightning.io.plr   markparametrizer   r   r   r   r   r   r   r   <module>   s$   


