<s> """<STR_LIT>""" <EOL> import unittest <EOL> import pyproct . data . handler . featurearray . test . data as data <EOL> from pyproct . data . handler . featurearray . featureArrayData import FeatureArrayData <EOL> import numpy <EOL> class TestFeatArrData ( unittest . TestCase ) : <EOL> @ classmethod <EOL> def setUpClass ( cls ) : <EOL> cls . data_path = data . __path__ [ <NUM_LIT:0> ] <EOL> def test_get_all_elements ( self ) : <EOL> expected_2_columns = [ [ <NUM_LIT> , <NUM_LIT> ] , [ <NUM_LIT> , <NUM_LIT> ] , [ <NUM_LIT> , <NUM_LIT> ] , [ <NUM_LIT> , <NUM_LIT> ] , [ <NUM_LIT> , <NUM_LIT> ] , [ <NUM_LIT> , <NUM_LIT> ] , [ <NUM_LIT> , <NUM_LIT> ] , [ <NUM_LIT> , <NUM_LIT> ] , [ <NUM_LIT> , <NUM_LIT> ] , [ <NUM_LIT> , <NUM_LIT> ] , [ <NUM_LIT> , <NUM_LIT> ] , [ <NUM_LIT> , <NUM_LIT> ] , [ <NUM_LIT> , <NUM_LIT> ] , [ <NUM_LIT> , <NUM_LIT> ] , [ <NUM_LIT> , <NUM_LIT> ] , [ <NUM_LIT> , <NUM_LIT> ] , [ <NUM_LIT> , <NUM_LIT> ] , [ <NUM_LIT:1.0> , <NUM_LIT> ] , [ <NUM_LIT> , <NUM_LIT> ] ] <EOL> expected_single_column = [ [ <NUM_LIT> ] , [ <NUM_LIT> ] , [ <NUM_LIT> ] , [ <NUM_LIT> ] , [ <NUM_LIT> ] , [ <NUM_LIT> ] , [ <NUM_LIT> ] , [ <NUM_LIT> ] , [ <NUM_LIT> ] , [ <NUM_LIT> ] , [ <NUM_LIT> ] , [ <NUM_LIT> ] , [ <NUM_LIT> ] , [ <NUM_LIT> ] , [ <NUM_LIT> ] , [ <NUM_LIT> ] , [ <NUM_LIT> ] , [ <NUM_LIT> ] , [ <NUM_LIT> ] ] <EOL> feature_data = FeatureArrayData ( data . expected_3 [ <NUM_LIT:0> ] ) <EOL> numpy . testing . assert_array_equal ( expected_2_columns , feature_data . get_all_elements ( ) ) <EOL> feature_data = FeatureArrayData ( data . expected_single_row ) <EOL> numpy . testing . assert_array_equal ( expected_single_column , feature_data . get_all_elements ( ) ) <EOL> if __name__ == "<STR_LIT:__main__>" : <EOL> unittest . main ( ) </s>
