test.py

import time
import sys
import os
from markov import Markov

start_time = time.time()

if len(sys.argv) > 1:
    seed = list(sys.argv[1:-2])
    depth = int(sys.argv[-2])
    iterations = int(sys.argv[-1])
else:
    seed = ['but', 'with', 'except', 'for', ':', 'app', 'band']
    depth = 4
    iterations = 1

dir = os.path.dirname(__file__)
filename = os.path.join(dir,
                        'facebook-page-post-scraper',
                        '1500321840185061_facebook_statuses.csv'
                        )

markov = Markov(filename, seed=seed, depth=depth)

for i in range(iterations):
    line = markov.generate_markov_text()
    while len(line) > 140:
        line = markov.generate_markov_text()
    print("\n" + line)

print("\n--- %s seconds ---" % (time.time() - start_time))