Skip to content

Commit

Permalink
Edited
Browse files Browse the repository at this point in the history
  • Loading branch information
gootra28 committed Dec 6, 2013
1 parent bef1eb8 commit 58f405f
Showing 1 changed file with 36 additions and 3 deletions.
39 changes: 36 additions & 3 deletions homework8/social_network.py
Original file line number Diff line number Diff line change
Expand Up @@ -43,18 +43,51 @@ def draw_practice_graph():

# Comment out this line after you have visually verified your practice graph.
# Otherwise, the picture will pop up every time that you run your program.
draw_practice_graph()
###draw_practice_graph()


###
### Problem 1b
###

# (Your code goes here.)
rj = nx.Graph()

rj.add_node("Mercutio")
rj.add_node("Capulet")
rj.add_node("Friar Laurence")
rj.add_node("Montague")
rj.add_node("Benvolio")
rj.add_node("Escalus")
rj.add_node("Paris")
rj.add_node("Romeo")
rj.add_node("Juliet")
rj.add_node("Nurse")
rj.add_node("Tybalt")

rj.add_edge("Nurse", "Juliet")
rj.add_edge("Juliet", "Capulet")
rj.add_edge("Juliet", "Tybalt")
rj.add_edge("Juliet", "Romeo")
rj.add_edge("Capulet", "Tybalt")
rj.add_edge("Capulet", "Escalus")
rj.add_edge("Capulet", "Paris")
rj.add_edge("Friar Laurence", "Romeo")
rj.add_edge("Friar Laurence", "Juliet")
rj.add_edge("Romeo", "Mercutio")
rj.add_edge("Mercutio", "Paris")
rj.add_edge("Mercutio", "Escalus")
rj.add_edge("Montague", "Escalus")
rj.add_edge("Benvolio", "Montague")
rj.add_edge("Romeo", "Montague")
rj.add_edge("Romeo", "Benvolio")
rj.add_edge("Escalus", "Paris")

assert len(rj.nodes()) == 11
assert len(rj.edges()) == 17




def draw_rj():
"""Draw the rj graph to the screen and to a file."""
nx.draw(rj)
Expand All @@ -64,7 +97,7 @@ def draw_rj():
# Comment out this line after you have visually verified your rj graph and
# created your PDF file.
# Otherwise, the picture will pop up every time that you run your program.
draw_rj()
###draw_rj()


###
Expand Down

0 comments on commit 58f405f

Please sign in to comment.