choice = raw_input("\n\nChoose:\n0. Exit\n1. Add a user\n2. Add a friend link\n3. Check connectivity\n4. Check degree of connection\n")
count = count+1
if choice=='0':
break
elif choice=='1':
username=raw_input("Enter a username:\t")
b = createUser(username,g,m)
if b.makeNode():
print "User created"
else:
print "User already exists"
elif choice=='2':
user1,user2 = raw_input("Enter the usernames of each user seperated by a tab:\t").split('\t')
c = createFriends(user1,user2,g,m)
if c.makeLinks():
print "Friendship link established"
else:
print "Either user doesn't exist or both names given are the same"
elif choice=='3':
print "Enter names seperated by Enter key. Press 0 to stop\n\n"
l=[]
while 1:
temp=raw_input("Enter name:\t")
if temp=='0':
break
l.append(temp)
d = checkConnectivity(l,g,m)
temp = d.check()
if temp==-1:
if b.makeNode():
print "Node added successfully"
else:
print "TEST CASE FAILED"
b = createUser("Steve Jobs",g,m)
if b.makeNode():
print "Node added successfully"
else:
print "TEST CASE FAILED"
print "\n\nTesting friendship creation:\n"
c = createFriends("Immanuel Kant","Elon Musk",g,m)
if c.makeLinks():
print "Link added successfully"
else:
print "TEST CASE FAILED"
c = createFriends("Immanuel Kant","Mark Twain",g,m)
if c.makeLinks():
print "Link added successfully"
else:
print "TEST CASE FAILED"
c = createFriends("Immanuel Kant","Steve Jobs",g,m)
if c.makeLinks():
print "Link added successfully"
else:
