Skip to content
Snippets Groups Projects
Select Git revision
  • fc55842d9d6814795a13793dd1d430b8447b2c62
  • master default protected
  • colorclipping
  • support-ragged
  • add_field_tests
  • support36
  • pyramid-master
  • 0.2.1
  • 0.2.0
  • 0.1.0
  • 0.0.1
  • 0.0.0
12 results

test_kernel.py

Blame
  • Forked from empyre / empyre
    357 commits behind the upstream repository.
    Jan Caron's avatar
    Jan Caron authored
    docstrings: updated!
    costfunction: CFAdapterScipyCG obsolete, deleted estimate_lambda()
    dataset: use_fftw and threads obsolete
    diagnostics: implemented stuff... much stuff...
    fft: Docstrings and implementation of threads
    magcreator: fixed bug in slab (dimension mix-up)
    phasemap: plot now also returns colorbar
    reconstruction: PrintIterator and optimize_simple_leastsq obsolete
    regularisator: Docstrings
    scripts: minor modifications
    scripts - reconstruct_random_pixels: now use optimize_linear
    testcases: analytic, compliance, kernel, magcreator, magdata, projector
    fc55842d
    History
    test_kernel.py 1.57 KiB
    # -*- coding: utf-8 -*-
    """Testcase for the magdata module."""
    
    
    import os
    import unittest
    
    import numpy as np
    from numpy.testing import assert_allclose
    
    from pyramid.kernel import Kernel
    
    
    class TestCaseKernel(unittest.TestCase):
    
        def setUp(self):
            self.path = os.path.join(os.path.dirname(os.path.realpath(__file__)), 'test_kernel')
            self.kernel = Kernel(1., dim_uv=(4, 4), b_0=1., geometry='disc')
    
        def tearDown(self):
            self.path = None
            self.kernel = None
    
        def test_kernel(self):
            np.save(os.path.join(self.path, 'ref_u'), self.kernel.u)
            np.save(os.path.join(self.path, 'ref_v'), self.kernel.v)
            np.save(os.path.join(self.path, 'ref_u_fft'), self.kernel.u_fft)
            np.save(os.path.join(self.path, 'ref_v_fft'), self.kernel.v_fft)
            ref_u = np.load(os.path.join(self.path, 'ref_u.npy'))
            ref_v = np.load(os.path.join(self.path, 'ref_v.npy'))
            ref_u_fft = np.load(os.path.join(self.path, 'ref_u_fft.npy'))
            ref_v_fft = np.load(os.path.join(self.path, 'ref_v_fft.npy'))
            assert_allclose(self.kernel.u, ref_u, err_msg='Unexpected behavior in u')
            assert_allclose(self.kernel.v, ref_v, err_msg='Unexpected behavior in v')
            assert_allclose(self.kernel.u_fft, ref_u_fft, err_msg='Unexpected behavior in u_fft')
            assert_allclose(self.kernel.v_fft, ref_v_fft, err_msg='Unexpected behavior in v_fft')
    
    
    if __name__ == '__main__':
        suite = unittest.TestLoader().loadTestsFromTestCase(TestCaseKernel)
        unittest.TextTestRunner(verbosity=2).run(suite)