<s> from sparselsh import LSH <EOL> from scipy . sparse import csr_matrix <EOL> import unittest <EOL> class IndexThenQueryTestCase ( unittest . TestCase ) : <EOL> """<STR_LIT>""" <EOL> def test_can_query_using_existing_row ( self ) : <EOL> X = csr_matrix ( [ <EOL> [ <NUM_LIT:3> , <NUM_LIT:0> , <NUM_LIT:0> , <NUM_LIT:0> , <NUM_LIT:0> , <NUM_LIT:0> , - <NUM_LIT:1> ] , <EOL> [ <NUM_LIT:0> , <NUM_LIT:1> , <NUM_LIT:0> , <NUM_LIT:0> , <NUM_LIT:0> , <NUM_LIT:0> , <NUM_LIT:1> ] , <EOL> [ <NUM_LIT:1> , <NUM_LIT:1> , <NUM_LIT:1> , <NUM_LIT:1> , <NUM_LIT:1> , <NUM_LIT:1> , <NUM_LIT:1> ] ] ) <EOL> y = [ "<STR_LIT:0>" , "<STR_LIT>" , "<STR_LIT>" ] <EOL> X_sim = csr_matrix ( [ [ <NUM_LIT:1> , <NUM_LIT:1> , <NUM_LIT:1> , <NUM_LIT:1> , <NUM_LIT:1> , <NUM_LIT:1> , <NUM_LIT:1> ] ] ) <EOL> lsh = LSH ( <NUM_LIT:4> , <EOL> X . shape [ <NUM_LIT:1> ] , <EOL> num_hashtables = <NUM_LIT:1> , <EOL> storage_config = { "<STR_LIT>" : None } ) <EOL> for ix in xrange ( X . shape [ <NUM_LIT:0> ] ) : <EOL> x = X . getrow ( ix ) <EOL> c = y [ ix ] <EOL> lsh . index ( x , extra_data = c ) <EOL> points = lsh . query ( X_sim , num_results = <NUM_LIT:1> ) <EOL> print points [ <NUM_LIT:0> ] <EOL> self . assertEqual ( type ( points [ <NUM_LIT:0> ] [ <NUM_LIT:0> ] [ <NUM_LIT:0> ] ) , csr_matrix ) <EOL> truth = points [ <NUM_LIT:0> ] [ <NUM_LIT:0> ] [ <NUM_LIT:0> ] . todense ( ) == X_sim . todense ( ) <EOL> self . assertTrue ( truth . all ( ) ) <EOL> if __name__ == '<STR_LIT:__main__>' : <EOL> unittest . main ( ) </s>
