def test_F6(): f = year_2008.F6() x = l.load_cec_auxiliary('F6_o', '2008') y = f(x) assert y == -140
def test_F1(): f = year_2008.F1() x = l.load_cec_auxiliary('F1_o', '2008') y = f(x) assert y == -450
def test_F4(): f = year_2008.F4() x = l.load_cec_auxiliary('F4_o', '2008') y = f(x) assert y == -330
def test_F5(): f = year_2008.F5() x = l.load_cec_auxiliary('F5_o', '2008') y = f(x) assert y == -180
def test_F10(): f = year_2010.F10(group_size=50) x = l.load_cec_auxiliary('F10_o', '2010') y = f(x) assert y == 0
def test_F3(): f = year_2008.F3() x = l.load_cec_auxiliary('F3_o', '2008') + 1 y = f(x) assert y == 390
def test_F3(): f = year_2010.F3() x = l.load_cec_auxiliary('F3_o', '2010') y = f(x) assert y == 0
def test_F8(): f = year_2013.F8() x = l.load_cec_auxiliary('F8_o', '2013') y = f(x) assert y == 0
def test_F14(): f = year_2013.F14() x = l.load_cec_auxiliary('F14_o', '2013') y = f(x) assert y >= 0
def test_F8(): f = year_2010.F8(group_size=50) x = l.load_cec_auxiliary('F8_o', '2010') x += 1 y = f(x) assert y == 950
def test_F20(): f = year_2010.F20() x = l.load_cec_auxiliary('F20_o', '2010') x += 1 y = f(x) assert round(y) == 0
def test_F18(): f = year_2010.F18(group_size=50) x = l.load_cec_auxiliary('F18_o', '2010') x += 1 y = f(x) assert round(y) == 0
def test_F13(): f = year_2010.F13(group_size=50) x = l.load_cec_auxiliary('F13_o', '2010') x += 1 y = f(x) assert y == 500
def test_F12(): f = year_2013.F12() x = l.load_cec_auxiliary('F12_o', '2013') x += 1 y = f(x) assert round(y) == 0
def _load_auxiliary_data(self, name, year, data): """Loads auxiliary data from a set of files. Args: name (str): Name of the function. year (str): Year of the function. data (list): List holding the variables to be loaded. """ # Iterates over the `data` list for dt in data: # Constructs the data file # Note that it will always be NAME_VARIABLE data_file = f'{name}_{dt}' # Loads the data to a temporary variable tmp = ld.load_cec_auxiliary(data_file, year) # Sets the temporary variable as a property setattr(self, dt, tmp)
def test_load_cec_auxiliary(): data = loader.load_cec_auxiliary('F1_o', '2005') assert data.shape[0] == 100