<s> """<STR_LIT>""" <EOL> import numpy as np <EOL> from pytest import raises <EOL> from . . _types import ( Bunch , _bunchify , _is_integer , _is_list , _is_float , <EOL> _as_list , _is_array_like , _as_array , _as_tuple , <EOL> _as_scalar , <EOL> ) <EOL> def test_bunch ( ) : <EOL> obj = Bunch ( ) <EOL> obj [ '<STR_LIT:a>' ] = <NUM_LIT:1> <EOL> assert obj . a == <NUM_LIT:1> <EOL> obj . b = <NUM_LIT:2> <EOL> assert obj [ '<STR_LIT:b>' ] == <NUM_LIT:2> <EOL> assert obj . copy ( ) == obj <EOL> def test_bunchify ( ) : <EOL> d = { '<STR_LIT:a>' : { '<STR_LIT:b>' : <NUM_LIT:0> } } <EOL> b = _bunchify ( d ) <EOL> assert isinstance ( b , Bunch ) <EOL> assert isinstance ( b [ '<STR_LIT:a>' ] , Bunch ) <EOL> def test_number ( ) : <EOL> assert not _is_integer ( None ) <EOL> assert not _is_integer ( <NUM_LIT> ) <EOL> assert _is_integer ( <NUM_LIT:3> ) <EOL> assert _is_integer ( np . arange ( <NUM_LIT:1> ) [ <NUM_LIT:0> ] ) <EOL> assert not _is_float ( None ) <EOL> assert not _is_float ( <NUM_LIT:3> ) <EOL> assert not _is_float ( np . array ( [ <NUM_LIT:3> ] ) [ <NUM_LIT:0> ] ) <EOL> assert _is_float ( <NUM_LIT> ) <EOL> assert _is_float ( np . array ( [ <NUM_LIT> ] ) [ <NUM_LIT:0> ] ) <EOL> def test_list ( ) : <EOL> assert not _is_list ( None ) <EOL> assert not _is_list ( ( ) ) <EOL> assert _is_list ( [ ] ) <EOL> assert _as_list ( None ) is None <EOL> assert _as_list ( <NUM_LIT:3> ) == [ <NUM_LIT:3> ] <EOL> assert _as_list ( [ <NUM_LIT:3> ] ) == [ <NUM_LIT:3> ] <EOL> assert _as_list ( ( <NUM_LIT:3> , ) ) == [ <NUM_LIT:3> ] <EOL> assert _as_list ( '<STR_LIT:3>' ) == [ '<STR_LIT:3>' ] <EOL> assert np . all ( _as_list ( np . array ( [ <NUM_LIT:3> ] ) ) == np . array ( [ <NUM_LIT:3> ] ) ) <EOL> def test_as_tuple ( ) : <EOL> assert _as_tuple ( <NUM_LIT:3> ) == ( <NUM_LIT:3> , ) <EOL> assert _as_tuple ( ( <NUM_LIT:3> , ) ) == ( <NUM_LIT:3> , ) <EOL> assert _as_tuple ( None ) is None <EOL> assert _as_tuple ( ( None , ) ) == ( None , ) <EOL> assert _as_tuple ( ( <NUM_LIT:3> , <NUM_LIT:4> ) ) == ( <NUM_LIT:3> , <NUM_LIT:4> ) <EOL> assert _as_tuple ( [ <NUM_LIT:3> ] ) == ( [ <NUM_LIT:3> ] , ) <EOL> assert _as_tuple ( [ <NUM_LIT:3> , <NUM_LIT:4> ] ) == ( [ <NUM_LIT:3> , <NUM_LIT:4> ] , ) <EOL> def test_as_scalar ( ) : <EOL> assert _as_scalar ( <NUM_LIT:1> ) == <NUM_LIT:1> <EOL> assert _as_scalar ( np . ones ( <NUM_LIT:1> ) [ <NUM_LIT:0> ] ) == <NUM_LIT:1.> <EOL> assert type ( _as_scalar ( np . ones ( <NUM_LIT:1> ) [ <NUM_LIT:0> ] ) ) == float <EOL> def test_array ( ) : <EOL> def _check ( arr ) : <EOL> assert isinstance ( arr , np . ndarray ) <EOL> assert np . all ( arr == [ <NUM_LIT:3> ] ) <EOL> _check ( _as_array ( <NUM_LIT:3> ) ) <EOL> _check ( _as_array ( <NUM_LIT> ) ) <EOL> _check ( _as_array ( [ <NUM_LIT:3> ] ) ) <EOL> _check ( _as_array ( <NUM_LIT:3> , np . float ) ) <EOL> _check ( _as_array ( <NUM_LIT> , np . float ) ) <EOL> _check ( _as_array ( [ <NUM_LIT:3> ] , np . float ) ) <EOL> _check ( _as_array ( np . array ( [ <NUM_LIT:3> ] ) ) ) <EOL> with raises ( ValueError ) : <EOL> _check ( _as_array ( np . array ( [ <NUM_LIT:3> ] ) , dtype = np . object ) ) <EOL> _check ( _as_array ( np . array ( [ <NUM_LIT:3> ] ) , np . float ) ) <EOL> assert _as_array ( None ) is None <EOL> assert not _is_array_like ( None ) <EOL> assert not _is_array_like ( <NUM_LIT:3> ) <EOL> assert _is_array_like ( [ <NUM_LIT:3> ] ) <EOL> assert _is_array_like ( np . array ( [ <NUM_LIT:3> ] ) ) </s>
