main.py 文件源码

def test_response_function(self):

        """
        Verify the response functions
        """
        for node in self.solver.G.nodes():

            self.assertIn("rf", self.solver.G.node[node])
            self.assertIsInstance(self.solver.G.node[node]["rf"], ResponseFunction )
            self.assertIs(type(self.solver.G.node[node]["rf"].resp_func(0.0)), np.float128)
            self.assertIs(type(self.solver.G.node[node]["rf"].resp_func(20.0)), np.float128)