import json
from pathlib import Path
from energy_validation import EnergyValidation

# load the dataset you want to simulate
data_file = open(Path('./data/VMTrial_cleaned.geojson').resolve(), 'r')
city = json.load(data_file)
buildings = city['features']

# input the usage type and quantities that you would like to simulate
test_batch = {
    '1000': 10,
    '4413': 10,
    '1921': 5
}

validate_meb_ep = EnergyValidation()
# if cleanup is True, removes all files in tmp directory
validate_meb_ep.run(building_set=buildings, building_quantities=test_batch, cleanup=False)