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