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 ) . METHOD_2 ( ) . METHOD_3 ( STRING_0 , STRING_1 ) . METHOD_4 ( STRING_2 , new org . IDENT_0 . spark . impl . IDENT_6 . IDENT_7 . Builder ( ) . METHOD_5 ( 4 ) . METHOD_6 ( 3 ) . build ( ) , STRING_0 ) . METHOD_4 ( "1" , new org . IDENT_0 . spark . impl . IDENT_6 . IDENT_7 . Builder ( ) . METHOD_5 ( 4 ) . METHOD_6 ( 3 ) . build ( ) , STRING_1 ) . METHOD_4 ( "2" , new org . IDENT_0 . spark . impl . IDENT_6 . IDENT_8 . Builder ( IDENT_9 . IDENT_10 . IDENT_11 ) . METHOD_5 ( 6 ) . METHOD_6 ( 3 ) . METHOD_7 ( IDENT_12 . IDENT_13 ) . build ( ) , STRING_2 , "1" ) . METHOD_8 ( "2" ) . build ( ) ; org . IDENT_0 . IDENT_1 . graph . IDENT_14 net = new org . IDENT_0 . IDENT_1 . graph . METHOD_9 ( conf ) ; net . init ( ) ; org . nd4j . linalg . dataset . api . iterator . IDENT_15 IDENT_16 = new org . IDENT_0 . IDENT_17 . iterator . impl . METHOD_10 ( 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_11 ( INT_1 ) ; while ( count < INT_0 ) { int IDENT_25 = ( r . METHOD_12 ( 5 ) ) + 1 ; if ( ( count + IDENT_25 ) > INT_0 ) IDENT_25 = INT_0 - count ; org . nd4j . linalg . api . ndarray . INDArray IDENT_26 = IDENT_19 . METHOD_13 ( ) . get ( org . nd4j . linalg . indexing . IDENT_27 . METHOD_14 ( count , ( count + IDENT_25 ) ) , org . nd4j . linalg . indexing . IDENT_27 . METHOD_15 ( ) ) ; expected . add ( net . METHOD_16 ( false , IDENT_26 , IDENT_26 ) ) ; IDENT_22 . add ( new IDENT_20 . METHOD_17 ( IDENT_23 , new org . nd4j . linalg . api . ndarray . INDArray [ ] { IDENT_26 , 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_18 ( IDENT_22 ) ; org . IDENT_0 . spark . impl . graph . IDENT_31 graph = new org . IDENT_0 . spark . impl . graph . METHOD_19 ( IDENT_30 , net , null ) ; org . IDENT_0 . spark . impl . IDENT_6 . Map < java . lang . Integer , org . nd4j . linalg . api . ndarray . INDArray [ ] > map = graph . METHOD_20 ( IDENT_29 , 16 ) . METHOD_21 ( ) ; 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_32 = map . get ( i ) [ 0 ] ; "<AssertPlaceHolder>" ; } } get ( int ) { return IDENT_33 . get ( i ) ; }
org . junit . Assert . assertEquals ( exp , IDENT_32 )