
    H>qjp                         d dl Z d dlmZ  e ed            Z e edd            Z e edd            Z G d d	e j                        Z	y)
    N)Event
   d   n         c                   6    e Zd Zd Zd Zd Zd Zd Zd Zd Z	y)	CombineTestc           	      N   t        j                  t        d      }t        j                  t        d      j	                  d      }|j                  |      }| j                  |j                         t        t        t              D cg c]  }|D ]  }|  c}}       y c c}}w N{Gz?intervalMbP?)	r   sequencearray1array2delaymergeassertEqualrunzip)selfe1e2eventjis         Q/opt/rentech/trading_bot/.venv/lib/python3.12/site-packages/tests/combine_test.py
test_mergezCombineTest.test_merge   s    ^^FT2^^FT288?66*'7A'7/0A'7'7 	8 '7s   B!c                 Z   t        j                  t        d      }t        j                  t        d      j	                  d      }t        j                  t
        d      j	                  d      }|j                  |||      }| j                  |j                         ddgt
        z          y )Nr   r   r   gMb`?r   r   )	r   r   r   r   r   array3switchr   r   )r   r   r   e3r   s        r   test_switchzCombineTest.test_switch   sy    ^^FT2^^FT288?^^FT288?		"b"%q#h&78    c                     t        j                  t        d      }t        j                  t        d      j	                  d      }|j                  |      }| j                  |j                         g d       y )Ng{Gz?r   gQ?)r            r   e   f   g   h   i   j   k   l   m   )r   r   r   r   r   concatr   r   r   r   r   r   s       r   test_concatzCombineTest.test_concat   sW    ^^FT2^^FT288>		" 'J 	Kr&   c                    t        j                  t        d      }t        j                  t        d      j	                  d      }|j                  ||      }| j                  |j                         t        t        z   t        z          y r   )r   r   r   r   r   chainr   r   r5   s       r   
test_chainzCombineTest.test_chain!   s\    ^^FT2^^FT288?R fvo&>?r&   c           	         t        j                  t              }t        j                  t              j	                  d      }|j                  |      }| j                  |j                         t        t        t        t                           y )Nr   )	r   r   r   r   r   r   r   r   listr5   s       r   test_zipzCombineTest.test_zip'   sX    ^^F#^^F#))%0r
d3vv+>&?@r&   c           	          t        j                  t              }|j                  |      }| j	                  |j                         t        t        t        t                           y )N)r   r   r   r   r   r   r;   )r   r   r   s      r   test_zip_selfzCombineTest.test_zip_self-   s?    ^^F#r
d3vv+>&?@r&   c                    t        j                  ddgd      }t        j                  ddgd      j                  d      }|j                  |      }| j	                  |j                         dt         j                  fdd	d
g       y )Nr   r(   r   r   r)   r*   r   )r   r)   )r(   r)   )r(   r*   )r   r   r   	ziplatestr   r   NO_VALUEr5   s       r   test_ziplatestzCombineTest.test_ziplatest2   so    ^^QFT2^^QFT288?R IIK1enn-vvvF	Hr&   N)
__name__
__module____qualname__r    r%   r6   r9   r<   r>   rB    r&   r   r
   r
   
   s*    89K@AA
Hr&   r
   )
unittesteventkitr   r;   ranger   r   r"   TestCaser
   rF   r&   r   <module>rK      sL     	eBi	eCo		eCo	-H(## -Hr&   