o
    Nis                     @   s   d Z ddlmZ ddlmZ ddlmZ ddlm  mZ ddl	m
Z
 ddlm  mZ e  G dd dejjZed	krDe
  dS dS )
zTests for moving_sequence.    )absolute_import)division)print_functionN)testingc                   @   s   e Zd Ze dd ZdS )MovingSequenceTestc                 C   s0  d\}}d}d| }dd|  }t jdt jd}t j||gt jd}|| || f}t jddgt jd}	tj||	|||d}
t |
jt j}
| j	| 
t j|
d	d
t |ft t |t jg  tt j|
dd
D ]*\}}|d }|||| ||| f }t |}| | 
||d |d   qkd S )N)   r            )r   r   r	   )dtyper   )start_positionvelocityoutput_sizesequence_length)r	   r
      )axis)tfonesuint8constantfloat32msimage_as_moving_sequencecastimage_sequenceassertAllEqualevaluate
reduce_sumfill	enumerateunstackassertGreaterEqual)selfhwr   vhvwimager   out_sizer   sequencei
full_imagejsubimagen_true r/   b/home/ubuntu/.local/lib/python3.10/site-packages/tensorflow_datasets/video/moving_sequence_test.pytest_images_as_moving_sequence!   s2   
 z1MovingSequenceTest.test_images_as_moving_sequenceN)__name__
__module____qualname__r   run_in_graph_and_eager_modesr1   r/   r/   r/   r0   r      s    r   __main__)__doc__
__future__r   r   r   tensorflow.compat.v2compatv2r   tensorflow_datasetsr   )tensorflow_datasets.video.moving_sequencevideomoving_sequencer   enable_v2_behaviortestTestCaser   r2   	test_mainr/   r/   r/   r0   <module>   s   "