<s> import unittest <EOL> import numpy as np <EOL> import deltasigma as ds <EOL> class TestDSQuantize ( unittest . TestCase ) : <EOL> """<STR_LIT>""" <EOL> def setUp ( self ) : <EOL> t = np . arange ( - <NUM_LIT:3> , <NUM_LIT:3> , <NUM_LIT> ) <EOL> t = t . reshape ( ( <NUM_LIT:1> , t . shape [ <NUM_LIT:0> ] ) ) <EOL> y = t <EOL> for _ in range ( <NUM_LIT:2> ) : <EOL> y = np . concatenate ( ( y , t ) , axis = <NUM_LIT:0> ) <EOL> n = np . array ( [ <NUM_LIT:2> , <NUM_LIT:3> , <NUM_LIT:4> ] ) <EOL> self . re1_first = ds . ds_quantize ( y , n ) <EOL> self . re1_second = ds . ds_quantize ( y [ <NUM_LIT:1> , : ] , n = <NUM_LIT:3> ) <EOL> self . re2 = np . array ( <EOL> [ [ - <NUM_LIT:1.> , - <NUM_LIT:1.> , - <NUM_LIT:1.> , - <NUM_LIT:1.> , - <NUM_LIT:1.> , - <NUM_LIT:1.> , - <NUM_LIT:1.> , - <NUM_LIT:1.> , - <NUM_LIT:1.> , - <NUM_LIT:1.> , - <NUM_LIT:1.> , <EOL> - <NUM_LIT:1.> , - <NUM_LIT:1.> , - <NUM_LIT:1.> , - <NUM_LIT:1.> , <NUM_LIT:1.> , <NUM_LIT:1.> , <NUM_LIT:1.> , <NUM_LIT:1.> , <NUM_LIT:1.> , <NUM_LIT:1.> , <NUM_LIT:1.> , <NUM_LIT:1.> , <NUM_LIT:1.> , <EOL> <NUM_LIT:1.> , <NUM_LIT:1.> , <NUM_LIT:1.> , <NUM_LIT:1.> , <NUM_LIT:1.> , <NUM_LIT:1.> ] , <EOL> [ - <NUM_LIT> , - <NUM_LIT> , - <NUM_LIT> , - <NUM_LIT> , - <NUM_LIT> , - <NUM_LIT> , - <NUM_LIT> , - <NUM_LIT> , - <NUM_LIT> , - <NUM_LIT> , <NUM_LIT:0.> , <NUM_LIT:0.> , <EOL> <NUM_LIT:0.> , <NUM_LIT:0.> , <NUM_LIT:0.> , <NUM_LIT:0.> , <NUM_LIT:0.> , <NUM_LIT:0.> , <NUM_LIT:0.> , <NUM_LIT:0.> , <NUM_LIT> , <NUM_LIT> , <NUM_LIT> , <NUM_LIT> , <NUM_LIT> , <NUM_LIT> , <EOL> <NUM_LIT> , <NUM_LIT> , <NUM_LIT> , <NUM_LIT> ] , <EOL> [ - <NUM_LIT> , - <NUM_LIT> , - <NUM_LIT> , - <NUM_LIT> , - <NUM_LIT> , - <NUM_LIT:1.> , - <NUM_LIT:1.> , - <NUM_LIT:1.> , - <NUM_LIT:1.> , - <NUM_LIT:1.> , - <NUM_LIT:1.> , - <NUM_LIT:1.> , <EOL> - <NUM_LIT:1.> , - <NUM_LIT:1.> , - <NUM_LIT:1.> , <NUM_LIT:1.> , <NUM_LIT:1.> , <NUM_LIT:1.> , <NUM_LIT:1.> , <NUM_LIT:1.> , <NUM_LIT:1.> , <NUM_LIT:1.> , <NUM_LIT:1.> , <NUM_LIT:1.> , <NUM_LIT:1.> , <NUM_LIT> , <EOL> <NUM_LIT> , <NUM_LIT> , <NUM_LIT> , <NUM_LIT> ] ] ) <EOL> def test_first_quantize ( self ) : <EOL> """<STR_LIT>""" <EOL> self . assertTrue ( <EOL> np . allclose ( self . re1_first , self . re2 , atol = <NUM_LIT> , rtol = <NUM_LIT> ) ) <EOL> def test_second_quantizer ( self ) : <EOL> """<STR_LIT>""" <EOL> self . assertTrue ( <EOL> np . allclose ( self . re1_second , self . re2 [ <NUM_LIT:1> , : ] , atol = <NUM_LIT> , rtol = <NUM_LIT> ) ) </s>
