# Path: machine/calc_correlation.py
# def find_correlation(movie_list, movie_for_correlation):
# def main(infile):
# 
# Path: machine/ml_main.py
# def main():


from machine import calc_correlation,ml_main

def test_calc_corr():
    corr_dict = calc_correlation.main("machine/ml_data1.py")
    assert (corr_dict['Terminator']['Terminator 2'] == 0.9986154372037583)


def test_ml_main():

# Path: machine/calc_correlation.py
# def find_correlation(movie_list, movie_for_correlation):
# def main(infile):
# 
# Path: machine/ml_main.py
# def main():


# Path: test_machine.py
from machine import calc_correlation,ml_main


def test_calc_corr():
    corr_dict = calc_correlation.main("machine/ml_data1.py")
    assert (corr_dict['Terminator']['Terminator 2'] == 0.9986154372037583)


def test_ml_main():

    rec = ml_main.main()