<s> from textplot . matrix import Matrix <EOL> def test_set_pair ( ) : <EOL> """<STR_LIT>""" <EOL> m = Matrix ( ) <EOL> m . set_pair ( '<STR_LIT:a>' , '<STR_LIT:b>' , <NUM_LIT:1> ) <EOL> assert m . get_pair ( '<STR_LIT:a>' , '<STR_LIT:b>' ) == <NUM_LIT:1> <EOL> assert m . get_pair ( '<STR_LIT:b>' , '<STR_LIT:a>' ) == <NUM_LIT:1> <EOL> def test_update_key_set ( ) : <EOL> """<STR_LIT>""" <EOL> m = Matrix ( ) <EOL> m . set_pair ( '<STR_LIT:a>' , '<STR_LIT:b>' , <NUM_LIT:1> ) <EOL> m . set_pair ( '<STR_LIT:a>' , '<STR_LIT:c>' , <NUM_LIT:2> ) <EOL> assert m . keys == set ( [ '<STR_LIT:a>' , '<STR_LIT:b>' , '<STR_LIT:c>' ] ) <EOL> def test_missing_key ( ) : <EOL> """<STR_LIT>""" <EOL> m = Matrix ( ) <EOL> m . set_pair ( '<STR_LIT:a>' , '<STR_LIT:b>' , <NUM_LIT:1> ) <EOL> assert m . get_pair ( '<STR_LIT:a>' , '<STR_LIT:c>' ) == None </s>
