import time
def topicchoice():
    print("Please select a topic that you wish to do a test on")
def login():
    print("login then")
def register():
    data = open("studentlogins.txt",'a+')
    print("register then")
    RegUsername = input("Please enter the first 3 letters of your of first name along with your age ")
    RegPassword = input("Please enter a password of your choice ")
    data.write(RegUsername + "," + RegPassword +"\n")
    data.close()

print("Welcome to Fergus' test")
def menu():
    print("Please select an option")
    print("1. Login")
    print("2. Register")
    option = input(": ")
    if option == "1":
        login()
    elif option == "2":
        register()
    else:
        print("That is not a valid option pls try again")
menu()