o i @sVddlZddlZddlmZddlmZmZdefddZddZ d d Z d d Z dS) N)TemporaryDirectory)ZipFilePathPointer open_datafilecontentscCsntj|d}tj||tjjd}t|d}|||Wd||fS1s.wY||fS)z Create a zip file under root_dir with a single file at rel_dir/file_name containing 'contents'. Return the full path to the zip file and the relative path inside the zip. z testdata.zip/wN)ospathjoinreplacesepzipfileZipFilewritestr)root_dirrel_dir file_namerzip_patharcnamezfrU/home/ubuntu/.local/lib/python3.10/site-packages/nltk/test/unit/test_open_datafile.py_create_test_zips rc CstR}tjdd}d}d}|d}t||||\}}|tjjdd}t||}t ||dd } | } Wdn1sBwY| |ksMJWddS1sXwYdS) zIopen_datafile should open a file inside a zip when given dir + file_name.corporatestpkg sample.txtzHello from zipped data utf-8rrencodingN rrr r encoderr r rrread) tmpdirrrtextdatarr dir_entryr fresultrrr2test_open_datafile_directory_and_filename_from_zips   "r(c CstV}tjdd}d}d}|d}t||||\}}|tjjdd}t||}||} t | dd } | } Wdn1sFwY| |ksQJWddS1s\wYdS) zJopen_datafile should open a file pointer directly when file_name is empty.rrrz%Direct file pointer from zipped data rr)rNr) r"rrr#r$rrr% dir_pointer file_pointerr&r'rrr(test_open_datafile_file_pointer_from_zip-s    "r+c CstV}tjdd}d}d}t||||\}}|tjjdd}t||}t||dd }| } Wdn1s=wYt | t t fsKJ| |ksQJWddS1s\wYdS)z?open_datafile should return a binary stream when encoding=None.rrz binary.binsrNr) rrr r rr r rrr! isinstancebytes bytearray) r"rr binary_datarrr%r)r&r'rrr'test_open_datafile_binary_mode_from_zipBs  "r0) rr tempfiler nltk.datarrr-rr(r+r0rrrrs