import sys from sys import stdin, stdout def seven(text): fpath = open('src/91K_nouns.txt') nouns = fpath.readlines() separated = text.split() #use nltk tokenize instead #print(separated) new_separated = [] for word in separated: word = word.lower() + '\n' if word in nouns: position = nouns.index(word) new_word = nouns[position + 7] #print(" replacing", new_position) new_separated.append(new_word.strip()) else: #print("notinlist") #print("adding to new_separated ", word) new_separated.append(word.strip()) #print(new_separated) return ' '.join(new_separated) text = stdin.read() output = seven(text) print(output) #sentence = input('What is your sentence? ') #print(seven(sentence)) # pytest requires that you name your tests with test_ # run with the 'pytest' command in your terminal #def test_seven(): # assert seven('Baboons') == 'babushkas' # assert seven('Baboons,') == 'babushkas'