def test_FindUsedOneDimensionalVerticalToExtend_emptyResult(self): """ 'ka' is a one dimensional variable but not y dimension. Only variables with 'y' dimension must be extended. """ expression = 'asd * ka' biggest_variable = R._get_biggest_variable(self.dic, expression) dims = biggest_variable.dims one_d_vars = R._find_used_one_dimensional_variables_to_extend(self.dic, dims, expression) self.assertEqual(list(), one_d_vars)
def test_FindUsedOneDimensionalVerticalToExtend_oneResult(self): """ 'ody' is a one dimensional variable with dimension 'y' """ expression = 'asd * ody + ka' biggest_variable = R._get_biggest_variable(self.dic, expression) dims = biggest_variable.dims one_d_vars = R._find_used_one_dimensional_variables_to_extend(self.dic, dims, expression) self.assertEqual(1, len(one_d_vars)) self.assertTrue('ody' in one_d_vars) variable = self.dic['ody'] self.assertEqual(('y',), variable.dims) self.assertEqual(17, variable.data[0])
def test_FindUsedOneDimensionalVerticalToExtend_twoResults(self): """ 'ody' and 'woody' are a one dimensional variables with dimension 'y' """ expression = 'asd * (ody - woody) + ka' biggest_variable = R._get_biggest_variable(self.dic, expression) dims = biggest_variable.dims one_d_vars = R._find_used_one_dimensional_variables_to_extend(self.dic, dims, expression) self.assertEqual(2, len(one_d_vars)) self.assertTrue('ody' in one_d_vars) self.assertTrue('woody' in one_d_vars) var_o = self.dic['ody'] self.assertEqual(('y',), var_o.dims) self.assertEqual(17, var_o.data[0]) var_w = self.dic['woody'] self.assertEqual(('y',), var_w.dims) self.assertEqual(18, var_w.data[0])