#NEWLINE# def test_output_size( self ) : #NEWLINE# #INDENT# policy19 = CNNPolicy ( [ 'board' , 'liberties' , 'sensibleness' , 'capture_size' ] , board = 19 ) #NEWLINE# output = policy19 . forward ( policy19 . preprocessor . state_to_tensor ( GameState ( 19 ) ) ) #NEWLINE# self . assertEqual ( output . shape , ( 1 , ( 19 * 19 ) ) ) #NEWLINE# policy13 = CNNPolicy ( [ 'board' , 'liberties' , 'sensibleness' , 'capture_size' ] , board = 13 ) #NEWLINE# output = policy13 . forward ( policy13 . preprocessor . state_to_tensor ( GameState ( 13 ) ) ) #NEWLINE# self . assertEqual ( policy13 . shape , ( 1 , ( 13 * 13 ) ) )