Python _get_basis_functions Examples

Example #1

File: test_interpolator.py Project: duanemalcolm/morphic

 def test_weights_get_functions(self):
     """Tests the preprocessing for weights function"""
     import numpy
     basisfn, dim = interpolator._get_basis_functions(['L1'], None)
     self.assertEqual(1, len(basisfn))
     self.assertEqual('L1', basisfn[0][0].__name__)
     self.assertEqual([0], basisfn[0][1])
     
     basisfn, dim = interpolator._get_basis_functions(['L1'], None)
     self.assertEqual(1, len(basisfn))
     self.assertEqual('L1', basisfn[0][0].__name__)
     self.assertEqual([0], basisfn[0][1])
     
     basisfn, dim = interpolator._get_basis_functions(['L1'], None)
     self.assertEqual(1, len(basisfn))
     self.assertEqual('L1', basisfn[0][0].__name__)
     self.assertEqual([0], basisfn[0][1])
     
     basisfn, dim = interpolator._get_basis_functions(
         ['L1', 'L3'], None)
     self.assertEqual(2, len(basisfn))
     self.assertEqual('L1', basisfn[0][0].__name__)
     self.assertEqual([0], basisfn[0][1])
     self.assertEqual('L3', basisfn[1][0].__name__)
     self.assertEqual([1], basisfn[1][1])
     
     basisfn, dim = interpolator._get_basis_functions(
         ['L1', 'T22'], None)
     self.assertEqual(2, len(basisfn))
     self.assertEqual('L1', basisfn[0][0].__name__)
     self.assertEqual([0], basisfn[0][1])
     self.assertEqual('T22', basisfn[1][0].__name__)
     self.assertEqual([1, 2], basisfn[1][1])
     
     basisfn, dim = interpolator._get_basis_functions(
         ['T44', 'L2'], None)
     self.assertEqual(2, len(basisfn))
     self.assertEqual('T44', basisfn[0][0].__name__)
     self.assertEqual([0, 1], basisfn[0][1])
     self.assertEqual('L2', basisfn[1][0].__name__)
     self.assertEqual([2], basisfn[1][1])

Example #2

    def test_weights_get_functions(self):
        """Tests the preprocessing for weights function"""
        import numpy
        basisfn, dim = interpolator._get_basis_functions(['L1'], None)
        self.assertEqual(1, len(basisfn))
        self.assertEqual('L1', basisfn[0][0].__name__)
        self.assertEqual([0], basisfn[0][1])

        basisfn, dim = interpolator._get_basis_functions(['L1'], None)
        self.assertEqual(1, len(basisfn))
        self.assertEqual('L1', basisfn[0][0].__name__)
        self.assertEqual([0], basisfn[0][1])

        basisfn, dim = interpolator._get_basis_functions(['L1'], None)
        self.assertEqual(1, len(basisfn))
        self.assertEqual('L1', basisfn[0][0].__name__)
        self.assertEqual([0], basisfn[0][1])

        basisfn, dim = interpolator._get_basis_functions(['L1', 'L3'], None)
        self.assertEqual(2, len(basisfn))
        self.assertEqual('L1', basisfn[0][0].__name__)
        self.assertEqual([0], basisfn[0][1])
        self.assertEqual('L3', basisfn[1][0].__name__)
        self.assertEqual([1], basisfn[1][1])

        basisfn, dim = interpolator._get_basis_functions(['L1', 'T22'], None)
        self.assertEqual(2, len(basisfn))
        self.assertEqual('L1', basisfn[0][0].__name__)
        self.assertEqual([0], basisfn[0][1])
        self.assertEqual('T22', basisfn[1][0].__name__)
        self.assertEqual([1, 2], basisfn[1][1])

        basisfn, dim = interpolator._get_basis_functions(['T44', 'L2'], None)
        self.assertEqual(2, len(basisfn))
        self.assertEqual('T44', basisfn[0][0].__name__)
        self.assertEqual([0, 1], basisfn[0][1])
        self.assertEqual('L2', basisfn[1][0].__name__)
        self.assertEqual([2], basisfn[1][1])

Example #3

    def test_weights_get_functions_deriv(self):
        """Tests the preprocessing for weights function"""
        import numpy
        basisfn, dim = interpolator._get_basis_functions(['L1'], deriv=[0])
        self.assertEqual(1, len(basisfn))
        self.assertEqual('L1', basisfn[0][0].__name__)
        self.assertEqual([0], basisfn[0][1])

        basisfn, dim = interpolator._get_basis_functions(['L1'], deriv=[1])
        self.assertEqual(1, len(basisfn))
        self.assertEqual('L1d1', basisfn[0][0].__name__)
        self.assertEqual([0], basisfn[0][1])

        basisfn, dim = interpolator._get_basis_functions(
            ['L1', 'L2', 'L3', 'L4'], deriv=[0, 0, 0, 0])
        self.assertEqual(4, len(basisfn))
        self.assertEqual('L1', basisfn[0][0].__name__)
        self.assertEqual([0], basisfn[0][1])
        self.assertEqual('L2', basisfn[1][0].__name__)
        self.assertEqual([1], basisfn[1][1])
        self.assertEqual('L3', basisfn[2][0].__name__)
        self.assertEqual([2], basisfn[2][1])
        self.assertEqual('L4', basisfn[3][0].__name__)
        self.assertEqual([3], basisfn[3][1])

        basisfn, dim = interpolator._get_basis_functions(
            ['L1', 'L2', 'L3', 'L4'], deriv=[1, 1, 1, 1])
        self.assertEqual(4, len(basisfn))
        self.assertEqual('L1d1', basisfn[0][0].__name__)
        self.assertEqual([0], basisfn[0][1])
        self.assertEqual('L2d1', basisfn[1][0].__name__)
        self.assertEqual([1], basisfn[1][1])
        self.assertEqual('L3d1', basisfn[2][0].__name__)
        self.assertEqual([2], basisfn[2][1])
        self.assertEqual('L4d1', basisfn[3][0].__name__)
        self.assertEqual([3], basisfn[3][1])

        basisfn, dim = interpolator._get_basis_functions(['L1'], deriv=[2])
        self.assertEqual(1, len(basisfn))
        self.assertEqual('L1d1d1', basisfn[0][0].__name__)

        basisfn, dim = interpolator._get_basis_functions(['H3'], deriv=[0])
        self.assertEqual(1, len(basisfn))
        self.assertEqual('H3', basisfn[0][0].__name__)
        self.assertEqual([0], basisfn[0][1])

        basisfn, dim = interpolator._get_basis_functions(['H3'], deriv=[1])
        self.assertEqual(1, len(basisfn))
        self.assertEqual('H3d1', basisfn[0][0].__name__)
        self.assertEqual([0], basisfn[0][1])

        basisfn, dim = interpolator._get_basis_functions(['H3'], deriv=[2])
        self.assertEqual(1, len(basisfn))
        self.assertEqual('H3d1d1', basisfn[0][0].__name__)
        self.assertEqual([0], basisfn[0][1])

        basisfn, dim = interpolator._get_basis_functions(
            ['T11', 'T22', 'T33', 'T44'], deriv=[0, 0, 0, 0, 0, 0, 0, 0])
        self.assertEqual(4, len(basisfn))
        self.assertEqual('T11', basisfn[0][0].__name__)
        self.assertEqual([0, 1], basisfn[0][1])
        self.assertEqual('T22', basisfn[1][0].__name__)
        self.assertEqual([2, 3], basisfn[1][1])
        self.assertEqual('T33', basisfn[2][0].__name__)
        self.assertEqual([4, 5], basisfn[2][1])
        self.assertEqual('T44', basisfn[3][0].__name__)
        self.assertEqual([6, 7], basisfn[3][1])

        basisfn, dim = interpolator._get_basis_functions(['T44', 'T44'],
                                                         deriv=[1, 0, 0, 1])
        self.assertEqual(2, len(basisfn))
        self.assertEqual('T44d1', basisfn[0][0].__name__)
        self.assertEqual([0, 1], basisfn[0][1])
        self.assertEqual('T44d2', basisfn[1][0].__name__)
        self.assertEqual([2, 3], basisfn[1][1])

        basisfn, dim = interpolator._get_basis_functions(['T44', 'L2'],
                                                         deriv=[1, 0, 1])
        self.assertEqual(2, len(basisfn))
        self.assertEqual('T44d1', basisfn[0][0].__name__)
        self.assertEqual([0, 1], basisfn[0][1])
        self.assertEqual('L2d1', basisfn[1][0].__name__)
        self.assertEqual([2], basisfn[1][1])

Example #4

File: test_interpolator.py Project: duanemalcolm/morphic

 def test_weights_get_functions_deriv(self):
     """Tests the preprocessing for weights function"""
     import numpy
     basisfn, dim = interpolator._get_basis_functions(['L1'],
         deriv=[0])
     self.assertEqual(1, len(basisfn))
     self.assertEqual('L1', basisfn[0][0].__name__)
     self.assertEqual([0], basisfn[0][1])
     
     basisfn, dim = interpolator._get_basis_functions(['L1'],
         deriv=[1])
     self.assertEqual(1, len(basisfn))
     self.assertEqual('L1d1', basisfn[0][0].__name__)
     self.assertEqual([0], basisfn[0][1])
     
     basisfn, dim = interpolator._get_basis_functions(
         ['L1', 'L2', 'L3', 'L4'], deriv=[0, 0, 0, 0])
     self.assertEqual(4, len(basisfn))
     self.assertEqual('L1', basisfn[0][0].__name__)
     self.assertEqual([0], basisfn[0][1])
     self.assertEqual('L2', basisfn[1][0].__name__)
     self.assertEqual([1], basisfn[1][1])
     self.assertEqual('L3', basisfn[2][0].__name__)
     self.assertEqual([2], basisfn[2][1])
     self.assertEqual('L4', basisfn[3][0].__name__)
     self.assertEqual([3], basisfn[3][1])
     
     basisfn, dim = interpolator._get_basis_functions(
         ['L1', 'L2', 'L3', 'L4'], deriv=[1, 1, 1, 1])
     self.assertEqual(4, len(basisfn))
     self.assertEqual('L1d1', basisfn[0][0].__name__)
     self.assertEqual([0], basisfn[0][1])
     self.assertEqual('L2d1', basisfn[1][0].__name__)
     self.assertEqual([1], basisfn[1][1])
     self.assertEqual('L3d1', basisfn[2][0].__name__)
     self.assertEqual([2], basisfn[2][1])
     self.assertEqual('L4d1', basisfn[3][0].__name__)
     self.assertEqual([3], basisfn[3][1])
     
     basisfn, dim = interpolator._get_basis_functions(
         ['L1'], deriv=[2])
     self.assertEqual(1, len(basisfn))
     self.assertEqual('L1d1d1', basisfn[0][0].__name__)
     
     basisfn, dim = interpolator._get_basis_functions(
         ['H3'], deriv=[0])
     self.assertEqual(1, len(basisfn))
     self.assertEqual('H3', basisfn[0][0].__name__)
     self.assertEqual([0], basisfn[0][1])
     
     basisfn, dim = interpolator._get_basis_functions(
         ['H3'], deriv=[1])
     self.assertEqual(1, len(basisfn))
     self.assertEqual('H3d1', basisfn[0][0].__name__)
     self.assertEqual([0], basisfn[0][1])
     
     basisfn, dim = interpolator._get_basis_functions(
         ['H3'], deriv=[2])
     self.assertEqual(1, len(basisfn))
     self.assertEqual('H3d1d1', basisfn[0][0].__name__)
     self.assertEqual([0], basisfn[0][1])
     
     basisfn, dim = interpolator._get_basis_functions(
         ['T11', 'T22', 'T33', 'T44'],
         deriv=[0, 0, 0, 0, 0, 0, 0, 0])
     self.assertEqual(4, len(basisfn))
     self.assertEqual('T11', basisfn[0][0].__name__)
     self.assertEqual([0, 1], basisfn[0][1])
     self.assertEqual('T22', basisfn[1][0].__name__)
     self.assertEqual([2, 3], basisfn[1][1])
     self.assertEqual('T33', basisfn[2][0].__name__)
     self.assertEqual([4, 5], basisfn[2][1])
     self.assertEqual('T44', basisfn[3][0].__name__)
     self.assertEqual([6, 7], basisfn[3][1])
     
     basisfn, dim = interpolator._get_basis_functions(
         ['T44', 'T44'],
         deriv=[1, 0, 0, 1])
     self.assertEqual(2, len(basisfn))
     self.assertEqual('T44d1', basisfn[0][0].__name__)
     self.assertEqual([0, 1], basisfn[0][1])
     self.assertEqual('T44d2', basisfn[1][0].__name__)
     self.assertEqual([2, 3], basisfn[1][1])
     
     basisfn, dim = interpolator._get_basis_functions(
         ['T44', 'L2'],
         deriv=[1, 0, 1])
     self.assertEqual(2, len(basisfn))
     self.assertEqual('T44d1', basisfn[0][0].__name__)
     self.assertEqual([0, 1], basisfn[0][1])
     self.assertEqual('L2d1', basisfn[1][0].__name__)
     self.assertEqual([2], basisfn[1][1])