METHOD_0 ( ) { org . IDENT_0 . IDENT_1 . conf . IDENT_2 conf = new org . IDENT_0 . IDENT_1 . conf . IDENT_3 . Builder ( ) . METHOD_1 ( IDENT_4 . IDENT_5 ) . list ( ) . METHOD_2 ( 0 , new org . IDENT_0 . spark . impl . IDENT_6 . IDENT_7 . Builder ( ) . METHOD_3 ( 4 ) . METHOD_4 ( 3 ) . build ( ) ) . METHOD_2 ( 1 , new org . IDENT_0 . spark . impl . IDENT_6 . IDENT_8 . Builder ( IDENT_9 . IDENT_10 . IDENT_11 ) . METHOD_3 ( 3 ) . METHOD_4 ( 3 ) . METHOD_5 ( IDENT_12 . IDENT_13 ) . build ( ) ) . build ( ) ; org . IDENT_0 . IDENT_1 . IDENT_6 . IDENT_14 net = new org . IDENT_0 . IDENT_1 . IDENT_6 . METHOD_6 ( conf ) ; net . init ( ) ; org . nd4j . linalg . dataset . api . iterator . IDENT_15 IDENT_16 = new org . IDENT_0 . IDENT_17 . iterator . impl . METHOD_7 ( INT_0 , INT_0 ) ; org . nd4j . linalg . dataset . api . IDENT_18 IDENT_19 = IDENT_16 . next ( ) ; org . IDENT_0 . spark . impl . IDENT_6 . List < org . nd4j . linalg . api . ndarray . INDArray > expected = new org . IDENT_0 . spark . impl . IDENT_6 . ArrayList ( ) ; org . IDENT_0 . spark . impl . IDENT_6 . List < IDENT_20 . IDENT_21 < java . lang . Integer , org . nd4j . linalg . api . ndarray . INDArray > > IDENT_22 = new org . IDENT_0 . spark . impl . IDENT_6 . ArrayList ( ) ; int count = 0 ; int IDENT_23 = 0 ; org . IDENT_0 . spark . impl . IDENT_6 . IDENT_24 r = new org . IDENT_0 . spark . impl . IDENT_6 . METHOD_8 ( INT_1 ) ; while ( count < INT_0 ) { int IDENT_25 = ( r . METHOD_9 ( 5 ) ) + 1 ; if ( ( count + IDENT_25 ) > INT_0 ) IDENT_25 = INT_0 - count ; org . nd4j . linalg . api . ndarray . INDArray IDENT_26 = IDENT_19 . METHOD_10 ( ) . get ( org . nd4j . linalg . indexing . IDENT_27 . METHOD_11 ( count , ( count + IDENT_25 ) ) , org . nd4j . linalg . indexing . IDENT_27 . METHOD_12 ( ) ) ; expected . add ( net . output ( IDENT_26 , false ) ) ; IDENT_22 . add ( new IDENT_20 . METHOD_13 ( IDENT_23 , IDENT_26 ) ) ; IDENT_23 ++ ; count += IDENT_25 ; } org . apache . spark . api . java . IDENT_28 < java . lang . Integer , org . nd4j . linalg . api . ndarray . INDArray > IDENT_29 = IDENT_30 . METHOD_14 ( IDENT_22 ) ; org . IDENT_0 . spark . impl . IDENT_6 . IDENT_31 IDENT_32 = new org . IDENT_0 . spark . impl . IDENT_6 . METHOD_15 ( IDENT_30 , net , null ) ; org . IDENT_0 . spark . impl . IDENT_6 . Map < java . lang . Integer , org . nd4j . linalg . api . ndarray . INDArray > map = IDENT_32 . METHOD_16 ( IDENT_29 , 16 ) . METHOD_17 ( ) ; for ( int i = 0 ; i < ( expected . size ( ) ) ; i ++ ) { org . nd4j . linalg . api . ndarray . INDArray exp = expected . get ( i ) ; org . nd4j . linalg . api . ndarray . INDArray IDENT_33 = map . get ( i ) ; "<AssertPlaceHolder>" ; } } get ( int ) { return IDENT_34 . get ( i ) ; }
org . junit . Assert . assertEquals ( exp , IDENT_33 )