<s> import numpy as np <EOL> from parakeet . testing_helpers import expect , run_local_tests <EOL> def create_const ( x ) : <EOL> return [ x , x , x , x ] <EOL> def test_create_const ( ) : <EOL> expect ( create_const , [ <NUM_LIT:1> ] , np . array ( [ <NUM_LIT:1> , <NUM_LIT:1> , <NUM_LIT:1> , <NUM_LIT:1> ] ) ) <EOL> expect ( create_const , [ <NUM_LIT:1.0> ] , np . array ( [ <NUM_LIT:1.0> , <NUM_LIT:1.0> , <NUM_LIT:1.0> , <NUM_LIT:1.0> ] ) ) <EOL> expect ( create_const , [ True ] , np . array ( [ True , True , True , True ] ) ) <EOL> def test_nested_2d ( ) : <EOL> expect ( create_const , [ np . array ( [ <NUM_LIT:1> , <NUM_LIT:2> ] ) ] , np . array ( [ [ <NUM_LIT:1> , <NUM_LIT:2> ] , [ <NUM_LIT:1> , <NUM_LIT:2> ] , [ <NUM_LIT:1> , <NUM_LIT:2> ] , [ <NUM_LIT:1> , <NUM_LIT:2> ] ] ) ) <EOL> expect ( create_const , [ np . array ( [ <NUM_LIT:1.0> , <NUM_LIT> ] ) ] , np . array ( [ [ <NUM_LIT:1.0> , <NUM_LIT> ] , [ <NUM_LIT:1.0> , <NUM_LIT> ] , [ <NUM_LIT:1.0> , <NUM_LIT> ] , [ <NUM_LIT:1.0> , <NUM_LIT> ] ] ) ) <EOL> expect ( create_const , [ np . array ( [ True , False ] ) ] , <EOL> np . array ( [ [ True , False ] , [ True , False ] , [ True , False ] , [ True , False ] ] ) ) <EOL> if __name__ == '<STR_LIT:__main__>' : <EOL> run_local_tests ( ) </s>
