Update apriori_test.py