<s> """<STR_LIT>""" <EOL> from __future__ import division , print_function , absolute_import <EOL> from numpy . testing import TestCase , assert_equal <EOL> from scipy . sparse import csr_matrix <EOL> import numpy as np <EOL> from scipy . sparse import extract <EOL> class TestExtract ( TestCase ) : <EOL> def setUp ( self ) : <EOL> self . cases = [ <EOL> csr_matrix ( [ [ <NUM_LIT:1> , <NUM_LIT:2> ] ] ) , <EOL> csr_matrix ( [ [ <NUM_LIT:1> , <NUM_LIT:0> ] ] ) , <EOL> csr_matrix ( [ [ <NUM_LIT:0> , <NUM_LIT:0> ] ] ) , <EOL> csr_matrix ( [ [ <NUM_LIT:1> ] , [ <NUM_LIT:2> ] ] ) , <EOL> csr_matrix ( [ [ <NUM_LIT:1> ] , [ <NUM_LIT:0> ] ] ) , <EOL> csr_matrix ( [ [ <NUM_LIT:0> ] , [ <NUM_LIT:0> ] ] ) , <EOL> csr_matrix ( [ [ <NUM_LIT:1> , <NUM_LIT:2> ] , [ <NUM_LIT:3> , <NUM_LIT:4> ] ] ) , <EOL> csr_matrix ( [ [ <NUM_LIT:0> , <NUM_LIT:1> ] , [ <NUM_LIT:0> , <NUM_LIT:0> ] ] ) , <EOL> csr_matrix ( [ [ <NUM_LIT:0> , <NUM_LIT:0> ] , [ <NUM_LIT:1> , <NUM_LIT:0> ] ] ) , <EOL> csr_matrix ( [ [ <NUM_LIT:0> , <NUM_LIT:0> ] , [ <NUM_LIT:0> , <NUM_LIT:0> ] ] ) , <EOL> csr_matrix ( [ [ <NUM_LIT:1> , <NUM_LIT:2> , <NUM_LIT:0> , <NUM_LIT:0> , <NUM_LIT:3> ] , [ <NUM_LIT:4> , <NUM_LIT:5> , <NUM_LIT:0> , <NUM_LIT:6> , <NUM_LIT:7> ] , [ <NUM_LIT:0> , <NUM_LIT:0> , <NUM_LIT:8> , <NUM_LIT:9> , <NUM_LIT:0> ] ] ) , <EOL> csr_matrix ( [ [ <NUM_LIT:1> , <NUM_LIT:2> , <NUM_LIT:0> , <NUM_LIT:0> , <NUM_LIT:3> ] , [ <NUM_LIT:4> , <NUM_LIT:5> , <NUM_LIT:0> , <NUM_LIT:6> , <NUM_LIT:7> ] , [ <NUM_LIT:0> , <NUM_LIT:0> , <NUM_LIT:8> , <NUM_LIT:9> , <NUM_LIT:0> ] ] ) . T , <EOL> ] <EOL> def find ( self ) : <EOL> for A in self . cases : <EOL> I , J , V = extract . find ( A ) <EOL> assert_equal ( A . toarray ( ) , csr_matrix ( ( ( I , J ) , V ) , shape = A . shape ) ) <EOL> def test_tril ( self ) : <EOL> for A in self . cases : <EOL> B = A . toarray ( ) <EOL> for k in [ - <NUM_LIT:3> , - <NUM_LIT:2> , - <NUM_LIT:1> , <NUM_LIT:0> , <NUM_LIT:1> , <NUM_LIT:2> , <NUM_LIT:3> ] : <EOL> assert_equal ( extract . tril ( A , k = k ) . toarray ( ) , np . tril ( B , k = k ) ) <EOL> def test_triu ( self ) : <EOL> for A in self . cases : <EOL> B = A . toarray ( ) <EOL> for k in [ - <NUM_LIT:3> , - <NUM_LIT:2> , - <NUM_LIT:1> , <NUM_LIT:0> , <NUM_LIT:1> , <NUM_LIT:2> , <NUM_LIT:3> ] : <EOL> assert_equal ( extract . triu ( A , k = k ) . toarray ( ) , np . triu ( B , k = k ) ) </s>
