import csv
import numpy as np
import trans_analysis as trans

def get_data(filename):
    return data

data = np.genfromtxt('trans_data.csv', delimiter=',')
trans.find_first_deficit(data)
# Should print:
# Found overdraft: transaction = 6  company = 1  balance =  -3.0

data = np.genfromtxt('trans_data2.csv', delimiter=',')
trans.find_first_deficit(data)
# Should print:
# Found overdraft: transaction = 17  company = 2  balance =  -7.0