def test_match_best_only(self):
df = extract([('test_data.xlsx', 18), ('test_data.xlsx', 19)],
fileout=None)
self.assertEqual(df.iloc[4, 3], 5)
def test_match_hierarchical_headers(self):
df = extract([('test_data.xlsx', 20), ('test_data.xlsx', 21)],
fileout=None)
self.assertEqual(df.iloc[3, 3], 7)
def test_depends_multiple_formulas(self):
df = extract([('test_data.xlsx', 15)], fileout=None)
self.assertEqual(df.iloc[1, 3], '=C2+B2')
def test_merge_dependent_formulas(self):
df = extract([('test_data.xlsx', 13), ('test_data.xlsx', 14)],
fileout=None)
self.assertEqual(df.iloc[1, 3], '=C2*10')
def test_multi_block_dep_merge(self):
df = extract([('test_data.xlsx', 11), ('test_data.xlsx', 13)],
fileout=None)
self.assertEqual(df.iloc[3, 2], '=A4+B4')
def test_split_range(self):
df = extract([('test_data.xlsx', 12)], fileout=None)
self.assertEqual(df.iloc[1, 2], '=A2+B2')
def test_side_formula(self):
df = extract([('test_data.xlsx', 10)], fileout=None)
self.assertEqual(df.iloc[1, 3], '=C2*10')
def test_multi_block_dep(self):
df = extract([('test_data.xlsx', 11)], fileout=None)
self.assertEqual(df.iloc[1, 2], '=A2+B2')
def test_extract_small(self):
extract([('test_data.xlsx', 3), ('test_data.xlsx', 4)], fileout=None)
def test_extract_lines_matches(self):
extract([('test_data.xlsx', 8), ('test_data.xlsx', 9)], fileout=None)
def test_heading(self):
df = extract([('test_data.xlsx', 6), ('test_data.xlsx', 7)],
fileout=None)
self.assertEqual(df.iloc[2, 0], 'a')
self.assertEqual(df.iloc[4, 1], 4)
def test_cut(self):
df = extract([('test_data.xlsx', 6)], fileout=None)
self.assertEqual(df.iloc[2, 0], 'a')
def test_extract(self):
extract([('test_data.xlsx', 0), ('test_data.xlsx', 1)], fileout=None)
def test_extract_single(self):
df = extract([('test_data.xlsx', 0)], fileout=None)
self.assertEqual(df.iloc[1, 1], 'SchoolID')
import sys
import time
from splyci.integration import extract
# TODO: Add usage instructions
if __name__ == '__main__':
fileout = sys.argv[1]
files = [(fn.split(':')[0], int(fn.split(':')[1])) for fn in sys.argv[2:]]
start = time.time()
print('start time:', start)
extract(files, fileout)
end = time.time()
print('end time:', end)
print('timediff: ', end - start)
