diff options
Diffstat (limited to 'weave/tests/test_slice_handler.py')
| -rw-r--r-- | weave/tests/test_slice_handler.py | 184 |
1 files changed, 184 insertions, 0 deletions
diff --git a/weave/tests/test_slice_handler.py b/weave/tests/test_slice_handler.py new file mode 100644 index 000000000..e951e6c1b --- /dev/null +++ b/weave/tests/test_slice_handler.py @@ -0,0 +1,184 @@ +import unittest +# Was getting a weird "no module named slice_handler error with this. + +from scipy_distutils.misc_util import add_grandparent_to_path, restore_path + +add_grandparent_to_path(__name__) +import slice_handler +from slice_handler import indexed_array_pattern +from ast_tools import * +restore_path() + +def print_assert_equal(test_string,actual,desired): + """this should probably be in scipy.scipy_test + """ + import pprint + try: + assert(actual == desired) + except AssertionError: + import cStringIO + msg = cStringIO.StringIO() + msg.write(test_string) + msg.write(' failed\nACTUAL: \n') + pprint.pprint(actual,msg) + msg.write('DESIRED: \n') + pprint.pprint(desired,msg) + raise AssertionError, msg.getvalue() + +class test_build_slice_atom(unittest.TestCase): + def generic_test(self,slice_vars,desired): + pos = slice_vars['pos'] + ast_list = slice_handler.build_slice_atom(slice_vars,pos) + actual = ast_to_string(ast_list) + print_assert_equal('',actual,desired) + def check_exclusive_end(self): + slice_vars = {'begin':'1', 'end':'2', 'step':'_stp', + 'single_index':'_index','pos':0} + desired = 'slice(1,2-1)' + self.generic_test(slice_vars,desired) + +class test_slice(unittest.TestCase): + + def generic_test(self,suite_string,desired): + import parser + ast_tuple = parser.suite(suite_string).totuple() + found, data = find_first_pattern(ast_tuple,indexed_array_pattern) + subscript = data['subscript_list'][1] #[0] is symbol, [1] is the supscript + actual = slice_handler.slice_ast_to_dict(subscript) + print_assert_equal(suite_string,actual,desired) + + def check_empty_2_slice(self): + """match slice from a[:]""" + test ="a[:]" + desired = {'begin':'_beg', 'end':'_end', 'step':'_stp', + 'single_index':'_index'} + self.generic_test(test,desired) + def check_begin_2_slice(self): + """match slice from a[1:]""" + test ="a[1:]" + desired = {'begin':'1', 'end':'_end', 'step':'_stp', + 'single_index':'_index'} + self.generic_test(test,desired) + def check_end_2_slice(self): + """match slice from a[:2]""" + test ="a[:2]" + desired = {'begin':'_beg', 'end':'2', 'step':'_stp', + 'single_index':'_index'} + self.generic_test(test,desired) + def check_begin_end_2_slice(self): + """match slice from a[1:2]""" + test ="a[1:2]" + desired = {'begin':'1', 'end':'2', 'step':'_stp', + 'single_index':'_index'} + self.generic_test(test,desired) + def check_empty_3_slice(self): + """match slice from a[::]""" + test ="a[::]" + desired = {'begin':'_beg', 'end':'_end', 'step':'_stp', + 'single_index':'_index'} + self.generic_test(test,desired) + def check_begin_3_slice(self): + """match slice from a[1::]""" + test ="a[1::]" + desired = {'begin':'1', 'end':'_end', 'step':'_stp', + 'single_index':'_index'} + self.generic_test(test,desired) + def check_end_3_slice(self): + """match slice from a[:2:]""" + test ="a[:2:]" + desired = {'begin':'_beg', 'end':'2', 'step':'_stp', + 'single_index':'_index'} + self.generic_test(test,desired) + def check_stp3_slice(self): + """match slice from a[::3]""" + test ="a[::3]" + desired = {'begin':'_beg', 'end':'_end', 'step':'3', + 'single_index':'_index'} + self.generic_test(test,desired) + def check_begin_end_3_slice(self): + """match slice from a[1:2:]""" + test ="a[1:2:]" + desired = {'begin':'1', 'end':'2','step':'_stp', + 'single_index':'_index'} + self.generic_test(test,desired) + def check_begin_step_3_slice(self): + """match slice from a[1::3]""" + test ="a[1::3]" + desired = {'begin':'1', 'end':'_end','step':'3', + 'single_index':'_index'} + self.generic_test(test,desired) + def check_end_step_3_slice(self): + """match slice from a[:2:3]""" + test ="a[:2:3]" + desired = {'begin':'_beg', 'end':'2', 'step':'3', + 'single_index':'_index'} + self.generic_test(test,desired) + def check_begin_end_stp3_slice(self): + """match slice from a[1:2:3]""" + test ="a[1:2:3]" + desired = {'begin':'1', 'end':'2', 'step':'3','single_index':'_index'} + self.generic_test(test,desired) + def check_expr_3_slice(self): + """match slice from a[:1+i+2:]""" + test ="a[:1+i+2:]" + desired = {'begin':'_beg', 'end':"1+i+2",'step':'_stp', + 'single_index':'_index'} + self.generic_test(test,desired) + def check_single_index(self): + """match slice from a[0]""" + test ="a[0]" + desired = {'begin':'_beg', 'end':"_end",'step':'_stp', + 'single_index':'0'} + self.generic_test(test,desired) + +def replace_whitespace(in_str): + import string + out = string.replace(in_str," ","") + out = string.replace(out,"\t","") + out = string.replace(out,"\n","") + return out + +class test_transform_slices(unittest.TestCase): + def generic_test(self,suite_string,desired): + import parser + ast_list = parser.suite(suite_string).tolist() + slice_handler.transform_slices(ast_list) + actual = ast_to_string(ast_list) + # Remove white space from expressions so that equivelant + # but differently formatted string will compare equally + import string + actual = replace_whitespace(actual) + desired = replace_whitespace(desired) + print_assert_equal(suite_string,actual,desired) + + def check_simple_expr(self): + """transform a[:] to slice notation""" + test ="a[:]" + desired = 'a[slice(_beg,_end,_stp)]' + self.generic_test(test,desired) + def check_simple_expr(self): + """transform a[:,:] = b[:,1:1+2:3] *(c[1-2+i:,:] - c[:,:])""" + test ="a[:,:] = b[:,1:1+2:3] *(c[1-2+i:,:] - c[:,:])" + desired = " a[slice(_beg,_end),slice(_beg,_end)] = "\ + " b[slice(_beg,_end), slice(1,1+2-1,3)] *"\ + " (c[slice(1-2+i,_end), slice(_beg,_end)] -"\ + " c[slice(_beg,_end), slice(_beg,_end)])" + self.generic_test(test,desired) + + +def test_suite(): + suites = [] + suites.append( unittest.makeSuite(test_slice,'check_') ) + suites.append( unittest.makeSuite(test_transform_slices,'check_') ) + suites.append( unittest.makeSuite(test_build_slice_atom,'check_') ) + total_suite = unittest.TestSuite(suites) + return total_suite + +def test(): + all_tests = test_suite() + runner = unittest.TextTestRunner() + runner.run(all_tests) + return runner + +if __name__ == "__main__": + test()
\ No newline at end of file |