o
    NiT	                     @   s   d Z ddlmZ ddlmZ ddlmZ ddlZddlmZ ddlm	Z
 ddlmZ dZG d	d
 d
ejZedkr?e  dS dS )z1Tests for tensorflow_datasets.text.cnn_dailymail.    )absolute_import)division)print_functionN)testing)cnn_dailymails}   Some article.
This is some article text.

@highlight

highlight text

@highlight

Highlight two

@highlight

highlight Three
c                   @   s6   e Zd ZejZddddZddddddZd	d
 ZdS )CnnDailymailTest      )train
validationtest zall_test.txtzall_train.txtzall_val.txt)cnn_stories
dm_stories	test_urls
train_urlsval_urlsc                 C   s   t jdd>}|t |  t|jtj	
d\}}| d| | d| t|jtj	
d\}}| d| W d    d S 1 sGw   Y  d S )NT)deletez1.0.0z(Some article. This is some article text.z2highlight text . Highlight two . highlight Three .z2.0.0z2highlight text .
Highlight two .
highlight Three .)tempfileNamedTemporaryFilewrite_STORY_FILEflushr   _get_art_absnametfdscoreVersionassertEqual)selffarticleabstract r#   h/home/ubuntu/.local/lib/python3.10/site-packages/tensorflow_datasets/summarization/cnn_dailymail_test.pytest_get_art_abs9   s"   


"z!CnnDailymailTest.test_get_art_absN)	__name__
__module____qualname__r   CnnDailymailDATASET_CLASSSPLITSDL_EXTRACT_RESULTr%   r#   r#   r#   r$   r   .   s    r   __main__)__doc__
__future__r   r   r   r   tensorflow_datasetsr   tensorflow_datasets.public_api
public_apir   !tensorflow_datasets.summarizationr   r   DatasetBuilderTestCaser   r&   	test_mainr#   r#   r#   r$   <module>   s   