def setUp(self):
variations = {
'a': {
'data_dir': 'data_dir_a'
},
'b': {
'data_dir': 'data_dir_b'
},
'c': {
'tag': 'tag_c'
},
'd': {},
}
self.worldline_set = analyze_worldline_set.WorldlineSet(
defaults, variations)
def setUp(self):
# Setup data
variations = {
'analyze_snum600': {
'tag': 'analyze_snum600'
},
'analyze_snum550': {
'tag': 'analyze_snum550'
},
}
self.w_set = analyze_worldline_set.WorldlineSet(defaults, variations)
self.stored_data_file = os.path.join(defaults['data_dir'],
'stored_quantity.hdf5')
if os.path.isfile(self.stored_data_file):
os.remove(self.stored_data_file)
def test_init(self, mock_constructor):
variations = {
'a': {
'data_dir': 'data_dir_a'
},
'b': {
'data_dir': 'data_dir_b'
},
'c': {
'tag': 'tag_c'
},
'd': {},
}
mock_constructor.side_effect = [
None,
] * len(variations)
calls = [
mock.call(data_dir='data_dir_a', tag=defaults['tag'], label='a'),
mock.call(data_dir='data_dir_b', tag=defaults['tag'], label='b'),
mock.call(data_dir=defaults['data_dir'], tag='tag_c', label='c'),
mock.call(data_dir=defaults['data_dir'],
tag=defaults['tag'],
label='d'),
]
worldline_set = analyze_worldline_set.WorldlineSet(
defaults, variations)
mock_constructor.assert_has_calls(calls, any_order=True)
# Check that it behaves like a dict
assert len(worldline_set) == len(variations)
for key in worldline_set.keys():
assert key in variations
for item in worldline_set:
assert item in variations