Method test_Auto

code_generators/interp_generator.py:744–843

(self)

       
    @property
    def test_Auto(self) -> str:
        # Try to write a test function that compares our function to NumPy.
        if type(self.py_Name) == type: # Ast  node
            funcName = self.py_Name.__name__.lower()
        else: # Function name
            funcName = self.py_Name
        
        required = None  # Do we require anything other than numpy for this test block?
        testCode = ['']
        evalFunc = None
        # There are a number of sample arrays in the autotest stub,
        # A_x, B_x, and C_x where 'x' is the dchar.
        
        
        for lib in self.libs:
            for I, retChar in enumerate( self.retFam ):
                required = 'np'  # The default module to test against is NumPy
                passedArgs = [ arg[I] for arg in self.argFams ]
                idArgs = [ arg.replace('?','1') for arg in passedArgs ]
                
                if funcName == 'cast' or funcName == 'copy':
                    return ''
                    
                if AUTOTEST_DICT[funcName]:
                    evalFunc = AUTOTEST_DICT[funcName]
                elif not hasattr( np, funcName ):
                    # Cycle through optional modules like scipy.special
                    for modName in OPTIONAL_TEST_MODULES:
                        try:
                            optMod = importlib.import_module(modName)
                            if hasattr( optMod, funcName ):
                                required = modName
                                # print( "Found {} in {}".format(funcName, optMod) ) 
                                break

                        except: # This changed from an ImportError <= 3.5 to a ModuleNotFoundError in 3.6
                            pass
                        
                    else:
                        if not funcName.startswith('unsafe'):
                            print( "Could not generate test for function {} in numpy, {}".format(funcName, OPTIONAL_TEST_MODULES))
                        return ''
                
                funcNameUnique = ''.join([funcName,'_',retChar.replace('?','1') ] + idArgs)
                # I wonder if it would be easier to build the program by hand
                # and not parse anything?  Or maybe I should put aliases 
                # in the OPTABLE?
                if evalFunc == None:
                    if len(idArgs) == 0:
                       evalFunc = '{0}()'.format(funcName)
                    elif len(idArgs) == 1:
                       evalFunc = '{0}(self.A_{1})'.format(funcName, idArgs[0])
                    elif len(idArgs) == 2:
                       evalFunc = '{0}(self.A_{1}, self.B_{2})'.format(funcName, idArgs[0], idArgs[1])     
                    elif len(idArgs) == 3:
                       evalFunc = '{0}(self.A_{1}, self.B_{2}, self.C_{3})'.format(funcName, idArgs[0], idArgs[1], idArgs[2])
                    externFunc = '.'.join( [required, evalFunc] )
                else: