def test_version_check(campaign):
info = CampaignInfo(
name='test2',
campaign_dir_prefix=default_campaign_prefix,
easyvvuq_version="some.other.version",
campaign_dir=str(campaign.tmp_path))
with pytest.raises(RuntimeError):
campaign2 = CampaignDB(location='sqlite:///{}/test.sqlite'.format(campaign.tmp_path),
new_campaign=True, name='test2', info=info)
info = CampaignInfo(
name='test3',
campaign_dir_prefix=default_campaign_prefix,
easyvvuq_version=uq.__version__,
campaign_dir=str(campaign.tmp_path))
campaign3 = CampaignDB(location='sqlite:///{}/test.sqlite'.format(campaign.tmp_path),
new_campaign=True, name='test3', info=info)
def campaign(tmp_path, app_info):
info = CampaignInfo(
name='test',
campaign_dir_prefix=default_campaign_prefix,
easyvvuq_version=uq.__version__,
campaign_dir=str(tmp_path))
campaign = CampaignDB(location='sqlite:///{}/test.sqlite'.format(tmp_path),
new_campaign=True, name='test', info=info)
campaign.tmp_path = str(tmp_path)
runs = [RunInfo('run', 'test', '.', 1, {'a': 1}, 1, 1) for _ in range(1010)]
run_names = ['Run_{}'.format(i) for i in range(1, 1011)]
campaign.add_runs(runs)
campaign.add_app(app_info)
return campaign
def init_db(self, name, work_dir='.'):
"""Initialize the connection with the database and either resume or create the campaign.
Parameters
----------
name: str
name of the campaign
work_dir: str
work directory, defaults to cwd
"""
self.campaign_db = CampaignDB(location=self.db_location)
if self.campaign_db.campaign_exists(name):
self.campaign_id = self.campaign_db.get_campaign_id(name)
self._active_app_name = self.campaign_db.get_active_app()[0].name
self.campaign_name = name
self._campaign_dir = self.campaign_db.campaign_dir(name)
if not os.path.exists(self._campaign_dir):
message = (
f"Campaign directory ({self.campaign_dir}) does not exist."
)
raise RuntimeError(message)
self._active_sampler_id = self.campaign_db.get_sampler_id(
self.campaign_id)
self._active_sampler = self.campaign_db.resurrect_sampler(
self._active_sampler_id)
self.set_app(self._active_app_name)
self.campaign_db.resume_campaign(name)
else:
if self._campaign_dir is None:
self._campaign_dir = tempfile.mkdtemp(prefix=name,
dir=work_dir)
info = CampaignInfo(name=name,
campaign_dir_prefix=default_campaign_prefix,
easyvvuq_version=easyvvuq.__version__,
campaign_dir=self._campaign_dir)
self.campaign_db.create_campaign(info)
self.campaign_name = name
self.campaign_id = self.campaign_db.get_campaign_id(
self.campaign_name)
def test_mv_collation(tmp_path, app_info):
mv_data = {
'timestep': [
1.0,
2.0,
3.0,
4.0,
5.0,
6.0,
7.0,
8.0,
9.0,
10.0,
11.0,
12.0,
13.0,
14.0,
15.0,
16.0,
17.0,
18.0,
19.0,
20.0,
21.0,
22.0,
23.0,
24.0,
25.0,
26.0,
27.0,
28.0,
29.0,
30.0,
31.0,
32.0,
33.0,
34.0,
35.0,
36.0,
37.0,
38.0,
39.0,
40.0,
41.0,
42.0,
43.0,
44.0,
45.0,
46.0,
47.0,
48.0,
49.0,
50.0,
51.0],
'time': [
5e-07,
1e-06,
1.5e-06,
2e-06,
2.5e-06,
3e-06,
3.5e-06,
4e-06,
4.5e-06,
5e-06,
5.5e-06,
6e-06,
6.5e-06,
7e-06,
7.5e-06,
8e-06,
8.5e-06,
9e-06,
9.5e-06,
1e-05,
1.05e-05,
1.1e-05,
1.15e-05,
1.2e-05,
1.25e-05,
1.3e-05,
1.35e-05,
1.4e-05,
1.45e-05,
1.5e-05,
1.55e-05,
1.6e-05,
1.65e-05,
1.7e-05,
1.75e-05,
1.8e-05,
1.85e-05,
1.9e-05,
1.95e-05,
2e-05,
2.05e-05,
2.1e-05,
2.15e-05,
2.2e-05,
2.25e-05,
2.3e-05,
2.35e-05,
2.4e-05,
2.45e-05,
2.5e-05,
2.5e-05],
'resulting_force': [
154334.0,
258961.0,
261115.0,
341040.0,
379343.0,
436974.0,
494847.0,
580487.0,
669071.0,
760177.0,
885250.0,
1003870.0,
1121100.0,
1240490.0,
1351730.0,
1441980.0,
1514690.0,
1562410.0,
1581580.0,
1569400.0,
1527670.0,
1462240.0,
1378540.0,
1284030.0,
1182790.0,
1086710.0,
996340.0,
920193.0,
867072.0,
840048.0,
838119.0,
862059.0,
910291.0,
975073.0,
1053810.0,
1141210.0,
1226020.0,
1303890.0,
1359100.0,
1387040.0,
1386630.0,
1353730.0,
1296010.0,
1224370.0,
1143670.0,
1061790.0,
986885.0,
925497.0,
879544.0,
858313.0,
862543.0]}
params = {
'temperature': 274.775390625,
'strain_rate': 0.004395019531250001,
'height': 0.0804453125}
info = CampaignInfo(
name='test',
campaign_dir_prefix=default_campaign_prefix,
easyvvuq_version=uq.__version__,
campaign_dir=str(tmp_path))
campaign = CampaignDB(location='sqlite:///{}/test.sqlite'.format(tmp_path),
new_campaign=True, name='test', info=info)
campaign.tmp_path = str(tmp_path)
runs = [RunInfo('run', 'test', '.', 1, params, 1, 1)]
run_names = ['Run_1']
campaign.add_runs(runs)
campaign.add_app(app_info)
results = [('Run_1', mv_data), ('Run_2', mv_data)]
campaign.store_results('test', results)
assert(not campaign.get_results('test').empty)
return campaign