124 lines
5.1 KiB
Python
124 lines
5.1 KiB
Python
from snakemake.io import expand
|
|
import os
|
|
import subprocess
|
|
import shutil
|
|
import yaml
|
|
|
|
def setUp():
|
|
# This utility setUp is intended to be run once before all tests are run
|
|
# It is intended the set up all the necessary fake data in order to test
|
|
# the rules and scipt files.
|
|
|
|
# Load the configuration file to get basic parameters
|
|
with open(r'tests/settings/testing_config.yaml') as file:
|
|
configs = yaml.full_load(file)
|
|
|
|
# Get the settings
|
|
pids = configs['PIDS']
|
|
|
|
|
|
# Reset the test data files
|
|
for pid in pids:
|
|
# Remove old data files if they exist
|
|
despath = os.path.join('data/raw/', pid)
|
|
if os.path.exists(despath) and os.path.isdir(despath):
|
|
shutil.rmtree(despath)
|
|
|
|
# Remove old processed files if they exist
|
|
propath = os.path.join('data/processed/', pid)
|
|
if os.path.exists(propath) and os.path.isdir(propath):
|
|
shutil.rmtree(propath)
|
|
|
|
# Create a fresh PID data directories necessary for this round of tests
|
|
os.mkdir(despath)
|
|
|
|
# Copy necessary data files
|
|
srcpath = os.path.join('tests/data/raw', pid)
|
|
srcfiles = os.listdir(srcpath)
|
|
for srcfile in srcfiles:
|
|
srcfile_path = os.path.join(srcpath, srcfile)
|
|
desfile = os.path.join(despath, srcfile)
|
|
shutil.copy(srcfile_path, desfile)
|
|
|
|
return configs
|
|
|
|
|
|
|
|
def generate_file_list(configs, sensor):
|
|
# Generates the list of files that would be produced for one sensor
|
|
# i.e. The sensor passed into the function.
|
|
|
|
# Initialize string of file path for both expected and actual metric values
|
|
act_str = "data/processed/{pid}/{sensor}_{sensor_type}{day_segment}.csv"
|
|
exp_str = "tests/data/processed/{pid}/{sensor}_{sensor_type}{day_segment}.csv"
|
|
|
|
sensor_cap = sensor.upper()
|
|
if 'DAY_SEGMENTS' and 'FEATURES' in configs[sensor_cap]:
|
|
sensor_type = []
|
|
if 'TYPES' in configs[sensor_cap]:
|
|
for each in configs[sensor_cap]['TYPES']:
|
|
sensor_type.append(each+'_')
|
|
|
|
act_file_list = expand(act_str,pid=configs["PIDS"],
|
|
sensor = sensor,
|
|
sensor_type = sensor_type,
|
|
day_segment = configs[sensor_cap]["DAY_SEGMENTS"])
|
|
|
|
exp_file_list = expand(exp_str,pid=configs["PIDS"],
|
|
sensor = sensor,
|
|
sensor_type = sensor_type,
|
|
day_segment = configs[sensor_cap]["DAY_SEGMENTS"])
|
|
|
|
return zip(act_file_list, exp_file_list)
|
|
|
|
|
|
def generate_sensor_file_lists(configs):
|
|
# Go through the configs and select those sensors with DAY_SEGMENTS,
|
|
# optionally TYPES then create expected files Return dictionary with
|
|
# list of file paths of expected and actual files for each sensor
|
|
# listed in the config file. Added for Travis.
|
|
|
|
# Initialize string of file path for both expected and actual metric values
|
|
act_str = "data/processed/{pid}/{sensor}_{sensor_type}{day_segment}.csv"
|
|
exp_str = "tests/data/processed/{pid}/{sensor}_{sensor_type}{day_segment}.csv"
|
|
|
|
# Get all the SENSORS in the config.yaml files
|
|
sensors = configs['SENSORS']
|
|
sensor_file_lists = {}
|
|
|
|
# Loop though all sensors and create the actual and expected file paths
|
|
for sensor in sensors:
|
|
if sensor == 'messages':
|
|
sensor = 'sms'
|
|
sensor_cap = 'SMS'
|
|
else:
|
|
sensor_cap = sensor.upper()
|
|
if 'DAY_SEGMENTS' in configs[sensor_cap]:
|
|
sensor_type = []
|
|
if 'TYPES' in configs[sensor_cap]:
|
|
for each in configs[sensor_cap]['TYPES']:
|
|
sensor_type.append(each+'_')
|
|
|
|
if sensor_type:
|
|
act_file_list = expand(act_str, pid=configs["PIDS"],
|
|
sensor = sensor,
|
|
sensor_type = sensor_type,
|
|
day_segment = configs[sensor_cap]["DAY_SEGMENTS"])
|
|
exp_file_list = expand(exp_str, pid=configs["PIDS"],
|
|
sensor = sensor,
|
|
sensor_type = sensor_type,
|
|
day_segment = configs[sensor_cap]["DAY_SEGMENTS"])
|
|
else:
|
|
act_file_list = expand(act_str, pid=configs["PIDS"],
|
|
sensor = sensor,
|
|
sensor_type = '',
|
|
day_segment = configs[sensor_cap]["DAY_SEGMENTS"])
|
|
exp_file_list = expand(exp_str, pid=configs["PIDS"],
|
|
sensor = sensor,
|
|
sensor_type = '',
|
|
day_segment = configs[sensor_cap]["DAY_SEGMENTS"])
|
|
|
|
sensor_file_lists[sensor_cap] = list(zip(act_file_list,exp_file_list))
|
|
|
|
return sensor_file_lists
|