Skip to content
Snippets Groups Projects
Commit ce31cb2d authored by mads's avatar mads
Browse files

new object oriented HTC file structure

parent b38ec982
No related branches found
No related tags found
No related merge requests found
Expand all Hide whitespace changes
Inline Side-by-side
wetb/hawc2/htc_file.py
+ 102
− 306
View file @ ce31cb2d
......@@ -7,241 +7,49 @@ See documentation of HTCFile below
'''
from collections import OrderedDict
import os
import time
import numpy as np
from wetb.functions.process_exec import pexec
class HTCSection(object):
section = []
def __init__(self, section):
self.section = section
def __str__(self):
return repr(self)
def __repr__(self):
return self.section2str(self.section)
def section2str(self, section, level=0):
#list(list(section.items())[5][1][0].items())[8]
s = ""
endcomment = ""
if isinstance(section, tuple):
section, endcomment = section
for k, vs in section.items():
for v, comment in ([vs], vs)[isinstance(vs, list)]:
if isinstance(v, OrderedDict):
s += "%sbegin %s;\t%s\n" % (" "*level, k, comment[0])
s += self.section2str(v, level + 1)
s += "%send %s;\t%s\n" % (" "*level, k, comment[1])
elif isinstance(v, list):
#s += "%sbegin %s;%s\n" % (" "*level, k, comment[0])
v = [_v.strip() for _v in v]
s += " "*(level) + ("\n%s" % (" "*(level))).join(v) + ("", "\n")[len(v) > 0]
#s += "%send %s;%s\n" % (" "*level, k, comment[1])
elif v is None:
pass
else:
if k.startswith(";"):
s += ("%s;\t%s" % (v, comment)).rstrip() + "\n"
else:
s += ("%s%s\t%s;\t%s" % (" "*(level), k, v, comment)).rstrip() + "\n"
return s + endcomment
def __getattribute__(self, *args, **kwargs):
try:
return object.__getattribute__(self, *args, **kwargs)
except:
#if args[0] not in self.section:
# return ""
if isinstance(self.section[args[0]][0], OrderedDict):
return HTCSection(self.section[args[0]][0])
return self.section[args[0]][0]
def __setattr__(self, *args, **kwargs):
k, v = args
if k in dir(self): # in ['section', 'filename', 'lines']:
return object.__setattr__(self, *args, **kwargs)
if k not in self.section:
if isinstance(v, (tuple, list)) and len(v) == 2:
self.section[k] = v
else:
self.section[k] = (v, "")
else:
if isinstance(v, (tuple, list)) and len(v) == 2:
self.section[k] = (str(v_) for v_ in v)
else:
comment = self.section[k][1]
self.section[args[0]] = (str(args[1]), comment)
def __delattr__(self, *args, **kwargs):
k, = args
if k in self.section:
del self.section[k]
def __eq__(self, other):
return str(self) == other
def __getitem__(self, section):
if "/" in section:
sections = section.split('/')
val = self.section[sections[0]][0]
for s in sections[1:]:
val = val[s][0]
return val
return self.section[section][0]
def add_section(self, name):
if name not in self.section:
self.section[name] = (OrderedDict(), ("", ""))
return HTCSection(self.section[name][0])
class HTCFile(HTCSection):
"""Htc file wrapper
Examples
--------
# load
>>> htcfile = HTCFile("test.htc")
# access (3 methods)
>>> print (htcfile['simulation']['time_stop'][0])
100
>>> print (htcfile['simulation/time_stop'])
100
>>> print (htcfile.simulation.time_stop)
100
# set values
>>> htcfile.simulation.time_stop = 200
# delete keys
>>> del htcfile.simulation.logfile
# safe key deleting
>>> try:
>>> del htcfile.hydro.water_properties.water_kinematics_dll
>>> except KeyError:
>>> pass
# save
>>> htcfile.save() # replace existing file
>>> htcfile.save('newfilename.htc')
# simulate
>>> htcfile.simulate('<hawc2path>/hawc2mb.exe')
"""
from wetb.hawc2.htc_contents import HTCContents, HTCSection, HTCLine, \
HTCDefaults
import os
class HTCFile(HTCContents, HTCDefaults):
filename = None
lines = []
level = 0
initial_comments = None
def __init__(self, filename=None):
self.contents = OrderedDict()
self.initial_comments = []
if filename is None:
self.filename = 'empty.htc'
self.lines = empty_htc.split("\n")
self.lines = self.empty_htc.split("\n")
else:
self.filename = filename
with open(filename) as fid:
self.lines = fid.readlines()
self.section = self.parse_section()
def set_name(self, name):
self.filename = "%s.htc" % name
self.simulation.logfile = "./log/%s.log" % name
self.output.filename = "./res/%s" % name
def parse_line(self):
global curr_line
line, *comments = self.lines[curr_line].split(";")
comments = ";".join(comments).rstrip()
while curr_line + 1 < len(self.lines) and self.lines[curr_line + 1].strip().startswith(";"):
curr_line += 1
comments += "\n%s" % self.lines[curr_line].rstrip()
return line, comments
def key_value(self, line):
if " " in line.strip() or "\t" in line.strip():
d = 9999
if " " in line.strip():
d = line.strip().index(" ")
if "\t" in line.strip() and line.strip().index('\t') < d:
d = line.strip().index('\t')
key = line.strip()[:d]
value = line.strip()[d + 1:]
return key, value
else:
return None, None
def parse_section(self, startline=0):
global curr_line
section = OrderedDict()
curr_line = startline
while curr_line < len(self.lines):
line, comments = self.parse_line()
if line.strip().lower().startswith("begin "):
key = line.strip()[6:]
if key == "output":
keys = ['filename', 'data_format', 'buffer', 'time']
sensors = []
output = OrderedDict()
for k in ['filename', 'data_format', 'buffer', 'time']:
output[k] = (None, "")
while not self.lines[curr_line + 1].lower().strip().startswith('end'):
curr_line += 1
line, comment = self.parse_line()
k, v = self.key_value(line)
if k in keys:
output[k] = (v, comment)
else:
sensors.append("%s;%s" % (line, comment))
curr_line += 1
line, endcomments = self.parse_line()
output['sensors'] = (sensors, ("", ""))
section[key] = (output, (comments, endcomments))
self.lines = [l.strip() for l in self.lines]
lines = self.lines.copy()
while lines:
if lines[0].startswith(";"):
self.initial_comments.append(lines.pop(0).strip())
elif lines[0].lower().startswith("begin"):
self._add_contents(HTCSection.from_lines(lines))
else:
line = HTCLine.from_lines(lines)
self._add_contents(line)
if line.name_ == "exit":
break
else:
curr_line += 1
value, end_comments = self.parse_section(curr_line)
while self.lines[curr_line + 1].strip().startswith(";"):
curr_line += 1
end_comments += "\n%s" % self.lines[curr_line].strip()
if key in section:
if not isinstance(section[key], list):
section[key] = [section[key], (value, (comments, end_comments))]
else:
section[key].append((value, (comments, end_comments)))
else:
section[key] = (value, (comments, end_comments))
elif line.lower().strip().startswith("end "):
return section, comments
elif line.lower().strip().startswith('exit'):
pass
elif " " in line.strip() or "\t" in line.strip():
key, value = self.key_value(line)
if key in section:
if not isinstance(section[key], list):
section[key] = [section[key], (value, comments)]
else:
section[key].append((value, comments))
else:
section[key] = (value, comments)
else:
section[';%d' % (curr_line + 1)] = (line, comments)
curr_line += 1
return section
def __setitem__(self, key, value):
self.contents[key] = value
def __str__(self):
return self.section2str(self.section) + "exit;"
return "".join(self.initial_comments + [c.__str__(1) for c in self])
def save(self, filename=None):
if filename is None:
......@@ -251,101 +59,89 @@ class HTCFile(HTCSection):
with open(filename, 'w') as fid:
fid.write(str(self))
def simulate(self, hawc2_path):
self.save(os.path.join(os.path.dirname(self.filename), "auto.htc"))
errorcode, stdout, stderr, cmd = pexec([hawc2_path, "./htc/auto.htc"], os.path.realpath("..", os.path.dirname(self.filename)))
def set_name(self, name):
self.filename = "%s.htc" % name
self.simulation.logfile = "./log/%s.log" % name
self.output.filename = "./res/%s" % name
if 'logfile' in self['simulation']:
with open(os.path.join(os.path.realpath(os.path.dirname(self.filename)), "../", self['simulation']['logfile'])) as fid:
log = fid.read()
def input_files(self):
files = [self.filename]
for mb in [self.new_htc_structure[mb] for mb in self.new_htc_structure.keys() if mb.startswith('main_body')]:
if "timoschenko_input" in mb:
files.append(mb.timoschenko_input.filename[0])
files.append(mb.get('external_bladedata_dll', [None, None, None])[2])
if 'aero' in self:
files.append(self.aero.ae_filename[0])
files.append(self.aero.pc_filename[0])
files.append(self.aero.get('external_bladedata_dll', [None, None, None])[2])
files.append(self.aero.get('output_profile_coef_filename', [None])[0])
if 'dynstall_ateflap' in self.aero:
files.append(self.aero.dynstall_ateflap.get('flap', [None] * 3)[2])
if 'bemwake_method' in self.aero:
files.append(self.aero.bemwake_method.get('a-ct-filename', [None] * 3)[0])
for dll in [self.dll[dll] for dll in self.get('dll', {}).keys()]:
files.append(dll.filename[0])
if 'wind' in self:
files.append(self.wind.get('user_defined_shear', [None])[0])
files.append(self.wind.get('wind.user_defined_shear_turbulence', [None])[0])
if 'wakes' in self:
files.append(self.wind.get('use_specific_deficit_file', [None])[0])
files.append(self.wind.get('write_ct_cq_file', [None])[0])
files.append(self.wind.get('write_final_deficits', [None])[0])
if 'hydro' in self:
if 'water_properties' in self.hydro:
files.append(self.hydro.water_properties.get('water_kinematics_dll', [None])[0])
if 'soil' in self:
if 'soil_element' in self.soil:
files.append(self.soil.soil_element.get('datafile', [None])[0])
if 'force' in self:
files.append(self.force.get('dll', [None])[0])
return [f for f in set(files) if f]
def output_files(self):
files = []
for k, index in [('simulation/logfile', 0),
('simulation/animation', 0),
('simulation/visualization', 0),
('new_htc_structure/beam_output_file_name', 0),
('new_htc_structure/body_output_file_name', 0),
('new_htc_structure/struct_inertia_output_file_name', 0),
('new_htc_structure/body_eigenanalysis_file_name', 0),
('new_htc_structure/constraint_output_file_name', 0),
('new_htc_structure/structure_eigenanalysis_file_name', 0),
('new_htc_structure/system_eigenanalysis', 0),
('output_at_time/filename', 0),
('turb_export/filename_u', 0),
('turb_export/filename_v', 0),
('turb_export/filename_w', 0)]:
line = self.get(k)
if line:
files.append(line[index])
dataformat = self.output.get('data_format', 'hawc_ascii')
res_filename = self.output.filename[0]
if dataformat == "gtsdf" or dataformat == "gtsdf64":
files.append(res_filename + ".hdf5")
elif dataformat == "flex_int":
files.append(res_filename + ".int")
files.append(os.path.join(os.path.dirname(res_filename), 'sensor'))
else:
log = stderr
if "error" in log.lower():
raise Exception ("Simulation failed: %s" % (log[:log.lower().index("error") + 1000]))
files.append(res_filename + ".sel")
files.append(res_filename + ".dat")
def add_sensor(self, sensor, nr=None):
if nr is None:
nr = len(self.output.sensors)
line, *comments = sensor.split(";")
comments = ";".join(comments).rstrip()
self.output.sensors.insert(nr, ("%s;\t%s" % (line.strip(), comments)).rstrip())
for key in [k for k in self.contents.keys() if k.startswith("output_at_time")]:
files.append(self[key]['filename'][0])
return [f for f in files if f]
def turbulence_files(self):
files = [self.get('wind.%s.filename_%s' % (type, comp), [None])[0] for type in ['mann', 'flex'] for comp in ['u', 'v', 'w']]
return [f for f in files if f]
def add_mann_turbulence(self, L=29.4, ae23=1, Gamma=3.9, seed=1001, high_frq_compensation=True,
filenames=None,
no_grid_points=(4096, 32, 32), box_dimension=(6000, 100, 100),
std_scaling=(1, .8, .5)):
wind = self.add_section('wind')
wind.turb_format = (1, "0=none, 1=mann,2=flex")
mann = wind.add_section('mann')
mann.create_turb_parameters = ("%.2f %.3f %.2f %d %d" % (L, ae23, Gamma, seed, high_frq_compensation), "L, alfaeps, gamma, seed, highfrq compensation")
if filenames is None:
filenames = ["./turb/turb_wsp%d_s%04d%s.bin" % (float(self.wind.wsp), seed, c) for c in ['u', 'v', 'w']]
if isinstance(filenames, str):
filenames = ["./turb/%s_s%04d%s.bin" % (filenames, seed, c) for c in ['u', 'v', 'w']]
for filename, c in zip(filenames, ['u', 'v', 'w']):
setattr(mann, 'filename_%s' % c, filename)
for c, n, dim in zip(['u', 'v', 'w'], no_grid_points, box_dimension):
setattr(mann, 'box_dim_%s' % c, "%d %.4f" % (n, dim / (n - 1)))
if std_scaling is None:
mann.dont_scale = 1
else:
try:
del mann.dont_scale
except KeyError:
pass
mann.std_scaling = "%f %f %f" % std_scaling
empty_htc = """begin simulation;
time_stop 600;
solvertype 1; (newmark)
on_no_convergence continue;
convergence_limits 1E3 1.0 1E-7; ; . to run again, changed 07/11
begin newmark;
deltat 0.02;
end newmark;
end simulation;
;
;----------------------------------------------------------------------------------------------------------------------------------------------------------------
;
begin new_htc_structure;
begin orientation;
end orientation;
begin constraint;
end constraint;
end new_htc_structure;
;
;----------------------------------------------------------------------------------------------------------------------------------------------------------------
;
begin wind ;
density 1.225 ;
wsp 10 ;
tint 1;
horizontal_input 1 ; 0=false, 1=true
windfield_rotations 0 0.0 0.0 ; yaw, tilt, rotation
center_pos0 0 0 -30 ; hub heigth
shear_format 1 0;0=none,1=constant,2=log,3=power,4=linear
turb_format 0 ; 0=none, 1=mann,2=flex
tower_shadow_method 0 ; 0=none, 1=potential flow, 2=jet
end wind;
;
;----------------------------------------------------------------------------------------------------------------------------------------------------------------
;
begin dll;
end dll;
;
;----------------------------------------------------------------------------------------------------------------------------------------------------------------
;
begin output;
general time;
end output;
exit;"""
if "__main__" == __name__:
f = HTCFile(r"C:\mmpe\hawc2\models\DTU10MWRef\htc\dtu_10mw_rwt.htc")
print (f.section2str(f['new_htc_structure']['main_body'][0]))
f.simulate("hawc2_path")
f = HTCFile(r"C:\mmpe\HAWC2\Hawc2_model\htc\NREL_5MW_reference_wind_turbine_launcher_test.htc")
print ("\n".join(f.output_files()))
wetb/hawc2/tests/test_files/test.htc deleted 100644 → 0
+ 0
− 773
View file @ b38ec982
This diff is collapsed.
wetb/hawc2/tests/test_htc_file.py
+ 107
− 30
View file @ ce31cb2d
......@@ -7,7 +7,7 @@ import os
import unittest
from datetime import datetime
from wetb.hawc2.htc_file import HTCFile
from wetb.hawc2.htc_file import HTCFile, HTCLine
os.chdir(os.path.relpath(".", __file__))
......@@ -20,15 +20,17 @@ class Test(unittest.TestCase):
def setUp(self):
unittest.TestCase.setUp(self)
self.testfilepath = "tests/test_files/"
self.testfilepath = "tests/test_files/htcfiles/"
def test_htc_file(self):
with open(self.testfilepath + 'test.htc') as fid:
def check_htc_file(self, f):
with open(f) as fid:
orglines = fid.readlines()
htcfile = HTCFile(self.testfilepath + "test.htc")
htcfile = HTCFile(f)
newlines = str(htcfile).split("\n")
#htcfile.save(self.testfilepath + 'tmp.htc')
htcfile.save(self.testfilepath + 'tmp.htc')
#with open(self.testfilepath + 'tmp.htc') as fid:
# newlines = fid.readlines()
......@@ -43,27 +45,41 @@ class Test(unittest.TestCase):
print ()
assert len(orglines) == len(newlines)
def test_htc_files(self):
for f in ['test3.htc']:
self.check_htc_file(self.testfilepath + f)
def test_htc_file_get(self):
htcfile = HTCFile(self.testfilepath + "test.htc")
self.assertEqual(htcfile['simulation']['logfile'][0], './logfiles/dlc12_iec61400-1ed3/dlc12_wsp10_wdir000_s1004.log')
self.assertEqual(htcfile['simulation/logfile'], './logfiles/dlc12_iec61400-1ed3/dlc12_wsp10_wdir000_s1004.log')
self.assertEqual(htcfile.simulation.logfile, './logfiles/dlc12_iec61400-1ed3/dlc12_wsp10_wdir000_s1004.log')
self.assertEqual(float(htcfile.simulation.newmark.deltat), 0.02)
self.assertEqual(htcfile.simulation.newmark, "deltat\t 0.02;\n")
s = """time_stop\t 100;\nsolvertype\t 1"""
htcfile = HTCFile(self.testfilepath + "test3.htc")
self.assertEqual(htcfile['simulation']['time_stop'][0], 200)
self.assertEqual(htcfile['simulation/time_stop'][0], 200)
self.assertEqual(htcfile['simulation.time_stop'][0], 200)
self.assertEqual(htcfile.simulation.time_stop[0], 200)
self.assertEqual(htcfile.dll.type2_dll.name[0], "risoe_controller")
self.assertEqual(htcfile.dll.type2_dll__2.name[0], "risoe_controller2")
s = """begin simulation;\n time_stop\t200;"""
self.assertEqual(str(htcfile.simulation)[:len(s)], s)
def test_htc_file_get2(self):
htcfile = HTCFile(self.testfilepath + "test.htc")
self.assertEqual(htcfile['simulation']['logfile'][0], './logfiles/dlc12_iec61400-1ed3/dlc12_wsp10_wdir000_s1004.log')
self.assertEqual(htcfile['simulation/logfile'][0], './logfiles/dlc12_iec61400-1ed3/dlc12_wsp10_wdir000_s1004.log')
self.assertEqual(htcfile['simulation.logfile'][0], './logfiles/dlc12_iec61400-1ed3/dlc12_wsp10_wdir000_s1004.log')
self.assertEqual(htcfile.simulation.logfile[0], './logfiles/dlc12_iec61400-1ed3/dlc12_wsp10_wdir000_s1004.log')
self.assertEqual(htcfile.simulation.newmark.deltat[0], 0.02)
def test_htc_file_set(self):
htcfile = HTCFile(self.testfilepath + "test.htc")
time_stop = int(htcfile.simulation.time_stop)
time_stop = htcfile.simulation.time_stop[0]
htcfile.simulation.time_stop = time_stop * 2
self.assertEqual(int(htcfile.simulation.time_stop), 2 * time_stop)
self.assertEqual(htcfile.simulation.time_stop[0], 2 * time_stop)
self.assertEqual(htcfile.simulation.time_stop.__class__, HTCLine)
def test_htc_file_set_key(self):
htcfile = HTCFile(self.testfilepath + "test.htc")
htcfile.simulation.name = "value"
self.assertEqual(htcfile.simulation.name, "value")
self.assertEqual(htcfile.simulation.name[0], "value")
def test_htc_file_del_key(self):
htcfile = HTCFile(self.testfilepath + "test.htc")
......@@ -84,20 +100,23 @@ class Test(unittest.TestCase):
def test_add_section(self):
htcfile = HTCFile()
htcfile.wind.add_section('mann')
htcfile.wind.mann.create_turb_parameters = ("29.4 1.0 3.9 1004 1.0", "L, alfaeps, gamma, seed, highfrq compensation")
self.assertTrue(htcfile.wind.mann.create_turb_parameters.startswith('29.4'))
htcfile.wind.mann.add_line("create_turb_parameters", [29.4, 1.0, 3.9, 1004, 1.0], "L, alfaeps, gamma, seed, highfrq compensation")
self.assertEqual(htcfile.wind.mann.create_turb_parameters[0], 29.4)
self.assertEqual(htcfile.wind.mann.create_turb_parameters[3], 1004)
self.assertEqual(htcfile.wind.mann.create_turb_parameters.comments, "L, alfaeps, gamma, seed, highfrq compensation")
def test_add_mann(self):
htcfile = HTCFile()
htcfile.add_mann_turbulence(30.1, 1.1, 3.3, 102, False)
s = """create_turb_parameters\t30.10 1.100 3.30 102 0;\tL, alfaeps, gamma, seed, highfrq compensation
filename_u\t./turb/turb_wsp10_s0102u.bin;
filename_v\t./turb/turb_wsp10_s0102v.bin;
filename_w\t./turb/turb_wsp10_s0102w.bin;
box_dim_u\t4096 1.4652;
box_dim_v\t32 3.2258;
box_dim_w\t32 3.2258;
std_scaling\t1.000000 0.800000 0.500000;"""
s = """begin mann;
create_turb_parameters\t30.1 1.1 3.3 102 0;\tL, alfaeps, gamma, seed, highfrq compensation
filename_u\t./turb/turb_wsp10_s0102u.bin;
filename_v\t./turb/turb_wsp10_s0102v.bin;
filename_w\t./turb/turb_wsp10_s0102w.bin;
box_dim_u\t4096 1.4652;
box_dim_v\t32 3.2258;
box_dim_w\t32 3.2258;
std_scaling\t1.000000 0.800000 0.500000;"""
for a, b in zip(s.split("\n"), str(htcfile.wind.mann).split("\n")):
self.assertEqual(a, b)
......@@ -105,14 +124,72 @@ std_scaling\t1.000000 0.800000 0.500000;"""
def test_sensors(self):
htcfile = HTCFile()
htcfile.set_name("test")
htcfile.add_sensor('wind free_wind 1 0 0 -30')
s = """filename\t./res/test;
general time;
wind free_wind 1 0 0 -30;"""
htcfile.output.add_sensor('wind', 'free_wind', [1, 0, 0, -30])
s = """begin output;
filename\t./res/test;
general time;
wind free_wind\t1 0 0 -30;"""
for a, b in zip(s.split("\n"), str(htcfile.output).split("\n")):
self.assertEqual(a, b)
#print (htcfile)
def test_output_at_time(self):
htcfile = HTCFile(self.testfilepath + "test2.htc")
self.assertTrue('begin output_at_time aero 15.0;' in str(htcfile))
def test_output_files(self):
htcfile = HTCFile(self.testfilepath + "test.htc")
output_files = htcfile.output_files()
#print (htcfile.output)
for f in ['./logfiles/dlc12_iec61400-1ed3/dlc12_wsp10_wdir000_s1004.log',
'./visualization/dlc12_wsp10_wdir000_s1004.hdf5',
'./animation/structure_aero_control_turb.dat',
'./res_eigen/dlc12_iec61400-1ed3/dlc12_wsp10_wdir000_s1004/dlc12_iec61400-1ed3/dlc12_wsp10_wdir000_s1004/dlc12_wsp10_wdir000_s1004_beam.dat',
'./res_eigen/dlc12_iec61400-1ed3/dlc12_wsp10_wdir000_s1004/dlc12_iec61400-1ed3/dlc12_wsp10_wdir000_s1004/dlc12_wsp10_wdir000_s1004_body.dat',
'./res_eigen/dlc12_iec61400-1ed3/dlc12_wsp10_wdir000_s1004/dlc12_iec61400-1ed3/dlc12_wsp10_wdir000_s1004/dlc12_wsp10_wdir000_s1004_struct.dat',
'./res_eigen/dlc12_iec61400-1ed3/dlc12_wsp10_wdir000_s1004/dlc12_iec61400-1ed3/dlc12_wsp10_wdir000_s1004/dlc12_wsp10_wdir000_s1004_body_eigen.dat',
'./res_eigen/dlc12_iec61400-1ed3/dlc12_wsp10_wdir000_s1004/dlc12_iec61400-1ed3/dlc12_wsp10_wdir000_s1004/dlc12_wsp10_wdir000_s1004_strc_eigen.dat',
'./res/dlc12_iec61400-1ed3/dlc12_wsp10_wdir000_s1004.sel',
'./res/dlc12_iec61400-1ed3/dlc12_wsp10_wdir000_s1004.dat',
'./res/rotor_check_inipos',
'./res/rotor_check_inipos2']:
try:
output_files.remove(f)
except ValueError:
raise ValueError(f + " is not in list")
self.assertFalse(output_files)
def test_turbulence_files(self):
htcfile = HTCFile(self.testfilepath + "dlc14_wsp10_wdir000_s0000.htc")
self.assertEqual(htcfile.turbulence_files(), ['./turb/turb_wsp10_s0000u.bin', './turb/turb_wsp10_s0000v.bin', './turb/turb_wsp10_s0000w.bin'])
def test_input_files(self):
htcfile = HTCFile(self.testfilepath + "test.htc")
input_files = htcfile.input_files()
#print (htcfile.output)
for f in ['./data/DTU_10MW_RWT_Tower_st.dat',
'./data/DTU_10MW_RWT_Towertop_st.dat',
'./data/DTU_10MW_RWT_Shaft_st.dat',
'./data/DTU_10MW_RWT_Hub_st.dat',
'data/2Bdown-rR1.08_blade_st.dat',
'./data/2Bdown-rR1.08_ae.dat',
'./data/2Bup_AEP095_pc.dat',
'./control/risoe_controller.dll',
'./control/generator_servo.dll',
'./control/mech_brake.dll',
'./control/servo_with_limits.dll',
'./control/towclearsens.dll',
'tests/test_files/htcfiles/test.htc'
]:
try:
input_files.remove(f)
except ValueError:
raise ValueError(f + " is not in list")
self.assertFalse(input_files)
if __name__ == "__main__":
#import sys;sys.argv = ['', 'Test.testName']
unittest.main()
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment