def test_generate_buildfile(self):
mydir = os.path.dirname(__file__)
path = os.path.join(mydir, 'data')
buildfilepath = os.path.join(path, 'build.yml')
assert not os.path.exists(buildfilepath)
build.generate_build_file(path)
assert os.path.exists(buildfilepath)
build.build_package('test_hdf5', 'generated', buildfilepath)
os.remove(buildfilepath)
def test_build():
"""
Test compilation
"""
mydir = os.path.dirname(__file__)
PATH = os.path.join(mydir, './build.yml')
build.build_package(USER, PACKAGE, PATH)
# TODO load DFs based on contents of .yml file at PATH
# not hardcoded vals (this will require loading modules from variable
# names, probably using __module__)
from quilt.data.test_121.groot import csv, tsv, xls
rows = len(csv.index)
assert rows == len(tsv.index) and rows == len(xls.index), \
'Expected dataframes to have same # rows'
cols = len(csv.columns)
print(csv.columns, xls.columns, tsv.columns)
assert cols == len(tsv.columns) and cols == len(xls.columns), \
'Expected dataframes to have same # columns'
def test_build_arrow(self):
"""
Test compilation
"""
os.environ["QUILT_PACKAGE_FORMAT"] = PackageFormat.ARROW.value
mydir = os.path.dirname(__file__)
path = os.path.join(mydir, './build.yml')
build.build_package('test_arrow', PACKAGE, path)
# TODO load DFs based on contents of .yml file at path
# not hardcoded vals (this will require loading modules from variable
# names, probably using __module__)
from quilt.data.test_arrow.groot import dataframes, README
csv = dataframes.csv
tsv = dataframes.csv
xls = dataframes.xls
rows = len(csv.index)
assert rows == len(tsv.index) and rows == len(xls.index), \
'Expected dataframes to have same # rows'
cols = len(csv.columns)
print(csv.columns, xls.columns, tsv.columns)
assert cols == len(tsv.columns) and cols == len(xls.columns), \
'Expected dataframes to have same # columns'
del os.environ["QUILT_PACKAGE_FORMAT"]
def test_failover(self):
mydir = os.path.dirname(__file__)
path = os.path.join(mydir, './build_failover.yml')
build.build_package('test_failover', PACKAGE, path)
from quilt.data.test_failover import bad