#NEWLINE# def test_extract( self ) : #NEWLINE# #INDENT# DF_expected = [ [ 1 , 1 , 1 , 1 , 1 , 1 ] , [ 1 , 0 , 1 , 0 , 1 , 0 ] , [ 1 , 0 , 1 , 0 , 1 , 1 ] , [ 1 , 0 , 0 , 1 , 1 , 0 ] , [ 1 , 0 , 0 , 1 , 0 , 1 ] ] #NEWLINE# DF = self . model . _TFIDFModel__extract_facts ( self . data . item_tags ) #NEWLINE# self . assertTrue ( np . array_equal ( DF . todense ( ) , DF_expected ) )