@requires_sklearn
def test_gat_plot_nonsquared():
 gat = _get_data(test_times=dict(start=0.0))
 gat.plot()
 ax = gat.plot_diagonal()
 scores = ax.get_children()[1].get_lines()[2].get_ydata()
 assert_equals(len(scores), len(gat.estimators_))

@ requires_sklearn def test_gat_plot_nonsquared ( ) : gat = _get_data ( test_times = dict ( start = 0.0 ) ) gat . plot ( ) ax = gat . plot_diagonal ( ) scores = ax . get_children ( ) [ 1 ] . get_lines ( ) [ 2 ] . get_ydata ( ) assert_equals ( len ( scores ) , len ( gat . estimators_ ) )
test gat diagonal plot .