<s> from mlxtend . tf_cluster . tf_base import _TfBaseCluster <EOL> import numpy as np <EOL> from mlxtend . utils import assert_raises <EOL> def test_init ( ) : <EOL> cl = _TfBaseCluster ( print_progress = <NUM_LIT:0> , random_seed = <NUM_LIT:1> ) <EOL> def test_check_array_1 ( ) : <EOL> X = np . array ( [ <NUM_LIT:1> , <NUM_LIT:2> , <NUM_LIT:3> ] ) <EOL> cl = _TfBaseCluster ( print_progress = <NUM_LIT:0> , random_seed = <NUM_LIT:1> ) <EOL> assert_raises ( ValueError , <EOL> '<STR_LIT>' , <EOL> cl . _check_array , <EOL> X ) <EOL> def test_check_array_2 ( ) : <EOL> X = list ( [ [ <NUM_LIT:1> ] , [ <NUM_LIT:2> ] , [ <NUM_LIT:3> ] ] ) <EOL> cl = _TfBaseCluster ( print_progress = <NUM_LIT:0> , random_seed = <NUM_LIT:1> ) <EOL> assert_raises ( ValueError , <EOL> '<STR_LIT>' , <EOL> cl . _check_array , <EOL> X ) <EOL> def test_check_array_3 ( ) : <EOL> X = np . array ( [ [ <NUM_LIT:1> ] , [ <NUM_LIT:2> ] , [ <NUM_LIT:3> ] ] ) <EOL> cl = _TfBaseCluster ( print_progress = <NUM_LIT:0> , random_seed = <NUM_LIT:1> ) <EOL> cl . _check_array ( X ) <EOL> def test_fit ( ) : <EOL> X = np . array ( [ [ <NUM_LIT:1> ] , [ <NUM_LIT:2> ] , [ <NUM_LIT:3> ] ] ) <EOL> tfr = _TfBaseCluster ( print_progress = <NUM_LIT:0> , random_seed = <NUM_LIT:1> ) <EOL> tfr . fit ( X ) <EOL> def test_predict_1 ( ) : <EOL> X = np . array ( [ [ <NUM_LIT:1> ] , [ <NUM_LIT:2> ] , [ <NUM_LIT:3> ] ] ) <EOL> cl = _TfBaseCluster ( print_progress = <NUM_LIT:0> , random_seed = <NUM_LIT:1> ) <EOL> assert_raises ( AttributeError , <EOL> '<STR_LIT>' , <EOL> cl . predict , <EOL> X ) <EOL> def test_predict_2 ( ) : <EOL> X = np . array ( [ [ <NUM_LIT:1> ] , [ <NUM_LIT:2> ] , [ <NUM_LIT:3> ] ] ) <EOL> cl = _TfBaseCluster ( print_progress = <NUM_LIT:0> , random_seed = <NUM_LIT:1> ) <EOL> cl . fit ( X ) <EOL> cl . predict ( X ) <EOL> def test_shuffle ( ) : <EOL> X = np . array ( [ [ <NUM_LIT:1> ] , [ <NUM_LIT:2> ] , [ <NUM_LIT:3> ] ] ) <EOL> y = np . array ( [ <NUM_LIT:1> , <NUM_LIT:2> , <NUM_LIT:3> ] ) <EOL> cl = _TfBaseCluster ( print_progress = <NUM_LIT:0> , random_seed = <NUM_LIT:1> ) <EOL> X_sh , y_sh = cl . _shuffle ( arrays = [ X , np . array ( y ) ] ) <EOL> np . testing . assert_equal ( X_sh , np . array ( [ [ <NUM_LIT:1> ] , [ <NUM_LIT:3> ] , [ <NUM_LIT:2> ] ] ) ) <EOL> np . testing . assert_equal ( y_sh , np . array ( [ <NUM_LIT:1> , <NUM_LIT:3> , <NUM_LIT:2> ] ) ) </s>
