Create word_count_test.py