analysis.py: test centrality options

sp00 · Oct 8, 2010 · a122229 · a122229
1 parent 183dfe9
commit a122229
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/analysis.py b/analysis.py
@@ -241,6 +241,11 @@ def main():
         if 'all' in centralities:
             centralities = 'betweenness,pagerank,degree'.split(',')
 
+        if set(centralities).difference(
+            'betweenness,pagerank,degree'.split(',')):
+            logging.error('Unknown centrality')
+            sys.exit(0)
+
         if "betweenness" in centralities:
             print >> sys.stderr, "betweenness"
             g.g.vs['bw'] = g.g.betweenness(weights='length', directed = True)