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])
