class BayesClassifier:
    def __init__(self, word_to_classify, catagories):
        pass

    def add_training_set(self, catagory, file):
        ''' Takes a catagory and the training corpus as file '''
        pass


    def classify(self, file):
        return "CATAGORY_IMPLEMENT_ME_PLEASE"


def test():
    bayes = BayesClassifier('Schimmel', ['Pferd','Brot','Klavier'])

    bayes.add_training_set('Klavier', 'Schimmel_Klavier')
    bayes.add_training_set('Pferd', 'Schimmel_Pferd')
    bayes.add_training_set('Brot', 'Schimmel_Brot')        

    print('Ergebnis wiki1 (Klavier): ', bayes.classify('Schimmel_wiki1.txt'))
    print('Ergebnis wiki2 (Pferd): ', bayes.classify('Schimmel_wiki2.txt'))      
    print('Ergebnis wiki3 (Pilz):',bayes.classify('Schimmel_wiki3.txt') )  

if __name__ == '__main__':
    test()