def test_inject_tariff():
gp = GamsParser('./test/gams/inject-siteanalysis.gms')
with open('./test/ao/scenario-context.json') as in_file, \
open('./test/ao/demand.json') as demand_file, \
open('./test/ao/nrel-sam-gen.json') as gen_file, \
open('./test/ao/tariff.json') as tariff_file:
data=json.load(in_file)
demand=json.load(demand_file)
gen=json.load(gen_file)
tariff=json.load(tariff_file)
print(tariff.keys())
print("Inject data")
d_map={
"nrel-sam-gen": gen,
"-LXKR3vVaHl1b3lDunJL": demand,
"-LXKLG1sepBvV6cTX0EN":tariff
}
new_model,inject_map=gp.inject(context=data,data=d_map)
print("Output")
#print(new_model)
inject_simple=[{
"id": item['item_id'],
"name": item['item_name'],
"type": item['item']['type'],
"rn": item['item']['rn']
} for item in inject_map]
pprint.pprint(inject_simple)
with open('tmp/output-siteanalysis.gms','w') as out_file:
out_file.write(new_model)
def test_inject_list():
gp = GamsParser('./test/gams/inject-basic.gms')
with open('./test/ao/scenario-context.json') as in_file:
data = json.load(in_file)
#print(data)
print("Inject data")
new_model, inject_map = gp.inject(context=data)
print("Output")
print(new_model)
def test_transform_real_shopmodel():
with open('./test/gams/real-shopmodel.gms', 'r') as in_file:
gp = GamsParser(in_file)
model = gp.transform()
print("\nModel\n")
print(model)
print("\nSymbols:\n")
for s in model.symbol():
print(s)
def test_transform_equation_def():
with open('./test/gams/equation-basic.gms', 'r') as in_file:
gp = GamsParser(in_file)
parse_tree = gp.parse()
print(parse_tree.pretty())
model = gp.transform()
print("\nModel\n")
print(model)
print("\nSymbols:\n")
for s in model.symbol():
print(s)
def test_transform_model_siteanalysis():
with open('./test/gams/real-siteanalysis.gms', 'r') as in_file:
gp = GamsParser(in_file)
model = gp.transform()
print("\nModel\n")
print(model)
print("\nSymbols:\n")
for s in model.symbol():
print(s)
dm = model.toJSON()
#print("\nmodel.toJSON\n")
#print(dm)
print("\nmodel.toDict Hack\n")
m = model.toDict()
scrub_meta(m)
pprint.pprint(m)
def test_transform_real_shopmodel():
with open('./test/gams/inject-datagen.inc','r') as gen_file, \
open('./test/gams/inject-datademand.inc','r') as demand_file, \
open('./test/gams/inject-siteanalysis.gms','r') as main_file:
gp_main = GamsParser(main_file)
gp_gen = GamsParser(gen_file)
gp_demand = GamsParser(demand_file)
model_main = gp_main.transform()
model_gen = gp_gen.transform()
model_demand = gp_gen.transform()
model = model_main + model_gen + model_demand
print("\nModel\n")
print(model)
print("\nSymbols:\n")
for s in model.symbol():
print(s)
def test_real_siteanalysis():
with open('./test/gams/real-siteanalysis.gms', 'r') as in_file:
gp = GamsParser(in_file)
parse_tree = gp.parse()
print(parse_tree.pretty())
def test_assignment_basic():
with open('./test/gams/assignment-basic.gms', 'r') as in_file:
gp = GamsParser(in_file)
parse_tree = gp.parse()
print(parse_tree.pretty())
def test_misc_comments():
with open('./test/gams/misc-comments.gms', 'r') as in_file:
gp = GamsParser(in_file)
parse_tree = gp.parse()
print(parse_tree.pretty())
def test_misc_include():
with open('./test/gams/misc-include.gms', 'r') as in_file:
gp = GamsParser(in_file)
parse_tree = gp.parse()
print(parse_tree.pretty())
def test_open_file():
gp = GamsParser('./test/gams/set-ranged.gms')
parse_tree = gp.parse()
print(parse_tree.pretty())
def test_equation_indexoperator():
with open('./test/gams/equation-indexoperator.gms', 'r') as in_file:
gp = GamsParser(in_file)
parse_tree = gp.parse()
print(parse_tree.pretty())
def test_scalar_basic():
with open('./test/gams/scalar-basic.gms', 'r') as in_file:
gp = GamsParser(in_file)
parse_tree = gp.parse()
print(parse_tree.pretty())
def test_real_eminit():
with open('./test/gams/real-eminit.gms', 'r') as in_file:
gp = GamsParser(in_file)
parse_tree = gp.parse()
print(parse_tree.pretty())
def test_equation_elementindex():
with open('./test/gams/equation-elementindex.gms', 'r') as in_file:
gp = GamsParser(in_file)
parse_tree = gp.parse()
print(parse_tree.pretty())
def test_set_multi():
with open('./test/gams/set-multi.gms', 'r') as in_file:
gp = GamsParser(in_file)
parse_tree = gp.parse()
print(parse_tree.pretty())
def test_equation_basic():
with open('./test/gams/equation-basic.gms', 'r') as in_file:
gp = GamsParser(in_file)
parse_tree = gp.parse()
print(parse_tree.pretty())
def test_parameter_tuple():
with open('./test/gams/parameter-tuple.gms', 'r') as in_file:
gp = GamsParser(in_file)
parse_tree = gp.parse()
print(parse_tree.pretty())
def test_real_shopmodel():
with open('./test/gams/real-shopmodel.gms', 'r') as in_file:
gp = GamsParser(in_file)
parse_tree = gp.parse()
print(parse_tree.pretty())
def test_variable_basic():
with open('./test/gams/variable-basic.gms', 'r') as in_file:
gp = GamsParser(in_file)
parse_tree = gp.parse()
print(parse_tree.pretty())
sys.exit(1)
############################
### Inject project data ####
############################
model_dir = '{sf}/model'.format(sf=task_handle.scenario_folder)
output_dir = '{sf}/output'.format(sf=task_handle.scenario_folder)
try:
os.makedirs(model_dir)
os.makedirs('{dir}/time'.format(dir=model_dir))
os.makedirs(output_dir)
except FileExistsError:
pass
gp = GamsParser('/var/task/template/siteanalysis.gms')
d_map = {"nrel-sam-gen": solar_data, f_tou_id: tou_data, f_load_id: load_data}
new_model, inject_map = gp.inject(context=task_handle.scenario_context,
data=d_map)
inject_map_file = '{dir}/inject_map.json'.format(dir=model_dir)
with open(inject_map_file, 'w') as out_file:
json.dump(inject_map, out_file, indent=2)
model_file = '{dir}/siteanalysis.gms'.format(dir=model_dir)
with open(model_file, 'w') as out_file:
out_file.write(new_model)
gp_gen = GamsParser('/var/task/template/gen.gms')
gen_model, im = gp_gen.inject(context=task_handle.scenario_context, data=d_map)