<s> from collections import defaultdict <EOL> from nose . tools import * <EOL> from nose import SkipTest <EOL> import networkx as nx <EOL> from networkx . algorithms . communicability_alg import * <EOL> class TestCommunicability : <EOL> @ classmethod <EOL> def setupClass ( cls ) : <EOL> global numpy <EOL> global scipy <EOL> try : <EOL> import numpy <EOL> except ImportError : <EOL> raise SkipTest ( '<STR_LIT>' ) <EOL> try : <EOL> import scipy <EOL> except ImportError : <EOL> raise SkipTest ( '<STR_LIT>' ) <EOL> def test_communicability ( self ) : <EOL> answer = { <NUM_LIT:0> : { <NUM_LIT:0> : <NUM_LIT> , <EOL> <NUM_LIT:1> : <NUM_LIT> <EOL> } , <EOL> <NUM_LIT:1> : { <NUM_LIT:0> : <NUM_LIT> , <EOL> <NUM_LIT:1> : <NUM_LIT> <EOL> } <EOL> } <EOL> result = communicability ( nx . path_graph ( <NUM_LIT:2> ) ) <EOL> for k1 , val in result . items ( ) : <EOL> for k2 in val : <EOL> assert_almost_equal ( answer [ k1 ] [ k2 ] , result [ k1 ] [ k2 ] , places = <NUM_LIT:7> ) <EOL> def test_communicability2 ( self ) : <EOL> answer_orig = { ( '<STR_LIT:1>' , '<STR_LIT:1>' ) : <NUM_LIT> , <EOL> ( '<STR_LIT:1>' , '<STR_LIT>' ) : <NUM_LIT> , <EOL> ( '<STR_LIT:1>' , '<STR_LIT>' ) : <NUM_LIT> , <EOL> ( '<STR_LIT:1>' , '<STR_LIT>' ) : <NUM_LIT> , <EOL> ( '<STR_LIT:1>' , '<STR_LIT>' ) : <NUM_LIT> , <EOL> ( '<STR_LIT>' , '<STR_LIT:1>' ) : <NUM_LIT> , <EOL> ( '<STR_LIT>' , '<STR_LIT>' ) : <NUM_LIT> , <EOL> ( '<STR_LIT>' , '<STR_LIT>' ) : <NUM_LIT> , <EOL> ( '<STR_LIT>' , '<STR_LIT>' ) : <NUM_LIT> , <EOL> ( '<STR_LIT>' , '<STR_LIT>' ) : <NUM_LIT> , <EOL> ( '<STR_LIT>' , '<STR_LIT:1>' ) : <NUM_LIT> , <EOL> ( '<STR_LIT>' , '<STR_LIT>' ) : <NUM_LIT> , <EOL> ( '<STR_LIT>' , '<STR_LIT>' ) : <NUM_LIT> , <EOL> ( '<STR_LIT>' , '<STR_LIT>' ) : <NUM_LIT> , <EOL> ( '<STR_LIT>' , '<STR_LIT>' ) : <NUM_LIT> , <EOL> ( '<STR_LIT>' , '<STR_LIT:1>' ) : <NUM_LIT> , <EOL> ( '<STR_LIT>' , '<STR_LIT>' ) : <NUM_LIT> , <EOL> ( '<STR_LIT>' , '<STR_LIT>' ) : <NUM_LIT> , <EOL> ( '<STR_LIT>' , '<STR_LIT>' ) : <NUM_LIT> , <EOL> ( '<STR_LIT>' , '<STR_LIT>' ) : <NUM_LIT> , <EOL> ( '<STR_LIT>' , '<STR_LIT:1>' ) : <NUM_LIT> , <EOL> ( '<STR_LIT>' , '<STR_LIT>' ) : <NUM_LIT> , <EOL> ( '<STR_LIT>' , '<STR_LIT>' ) : <NUM_LIT> , <EOL> ( '<STR_LIT>' , '<STR_LIT>' ) : <NUM_LIT> , <EOL> ( '<STR_LIT>' , '<STR_LIT>' ) : <NUM_LIT> } <EOL> answer = defaultdict ( dict ) <EOL> for ( k1 , k2 ) , v in answer_orig . items ( ) : <EOL> answer [ k1 ] [ k2 ] = v <EOL> G1 = nx . Graph ( [ ( '<STR_LIT>' , '<STR_LIT>' ) , ( '<STR_LIT>' , '<STR_LIT>' ) , ( '<STR_LIT>' , '<STR_LIT>' ) , <EOL> ( '<STR_LIT>' , '<STR_LIT>' ) , ( '<STR_LIT>' , '<STR_LIT:1>' ) , ( '<STR_LIT>' , '<STR_LIT>' ) ] ) <EOL> result = communicability ( G1 ) <EOL> for k1 , val in result . items ( ) : <EOL> for k2 in val : <EOL> assert_almost_equal ( answer [ k1 ] [ k2 ] , result [ k1 ] [ k2 ] , places = <NUM_LIT:7> ) <EOL> result = communicability_exp ( G1 ) <EOL> for k1 , val in result . items ( ) : <EOL> for k2 in val : <EOL> assert_almost_equal ( answer [ k1 ] [ k2 ] , result [ k1 ] [ k2 ] , places = <NUM_LIT:7> ) </s>
