import grocery_analytics as groc

def test(csv_file, min_occurence, printall):
    # Extract the data
    data = groc.get_data(csv_file)

    # Get the list of cities and print
    cities = groc.get_cities(data)
    print('-------- cities -----------------------')
    print(cities)

    # Get the list of products and print
    print('-------- products ---------------------')
    products = groc.get_products(data)
    products = groc.sort_products(products)
    print('Number of products:', len(products))
    if printall:
        print(products)

    # Extract the transactions in list form
    print('-------- transactions -----------------')
    transactions = groc.get_purchases(data)
    print('Number of transactions:', len(transactions))
    if printall:
        for row in transactions:
            print(row)

    # Compute support
    support = groc.compute_support(products, transactions, min_occurence)
    sorted_support = groc.sort_support(support)
    print('-------- support ----------------------')
    for row in sorted_support:
        print(row)


# Test 1
test('shopping_data_small.csv', 1, True)

# Test 2
#test('shopping_data_medium.csv', 300, False)

# Test 3
#test('shopping_data_full.csv', 1000, False)