def test_create(self): register = NDimensionalRegister() assert register.names == [] with raises(ValueError) as ex: register.get_entry('nonexistent') assert "ResolutionSet 'nonexistent' not registered" in str(ex.value)
def test_months_load(self, months): """Pass a monthly time-interval definition into the register """ register = NDimensionalRegister() register.register(IntervalSet('months', months)) actual = register.get_entry('months') expected = [ Interval('jan', ('P0M', 'P1M')), Interval('feb', ('P1M', 'P2M')), Interval('mar', ('P2M', 'P3M')), Interval('apr', ('P3M', 'P4M')), Interval('may', ('P4M', 'P5M')), Interval('jun', ('P5M', 'P6M')), Interval('jul', ('P6M', 'P7M')), Interval('aug', ('P7M', 'P8M')), Interval('sep', ('P8M', 'P9M')), Interval('oct', ('P9M', 'P10M')), Interval('nov', ('P10M', 'P11M')), Interval('dec', ('P11M', 'P12M')) ] for idx, interval in enumerate(expected): assert actual.data[idx] == interval
def test_remap_interval_load(self, remap_months): register = NDimensionalRegister() intervals = IntervalSet('remap_months', remap_months) register.register(intervals) actual = register.get_entry('remap_months') assert actual == intervals
def test_interval_loads(self): """Pass a time-interval definition into the register """ data = [{'name': '1_1', 'interval': [('PT0H', 'PT1H')]}] register = NDimensionalRegister() register.register(IntervalSet('energy_supply_hourly', data)) assert register.names == ['energy_supply_hourly'] actual = register.get_entry('energy_supply_hourly') element = Interval('1_1', ('PT0H', 'PT1H'), base_year=2010) expected = [] expected.append(element) assert actual.data == expected