test_runner_planar.py

import os, time, datetime, warnings
import numpy as np
import matplotlib.pyplot as plt
import matplotlib.mlab as mlab

from math import pi, log

from termcolor import colored, cprint

from scipy.io import loadmat, savemat
from scipy.sparse import spdiags, eye, bmat

from scipy.sparse.linalg import spsolve, factorized, splu
from scipy.linalg import solve, lu

from scipy.io import loadmat, savemat

from scipy.sparse import SparseEfficiencyWarning

# suppress warnings
warnings.simplefilter("ignore", SparseEfficiencyWarning)
warnings.filterwarnings("ignore", category=FutureWarning)

import WAKZK_planar, WAKZK, WAKZK_Gaussian
import SourceFilterH, SynthAxScan, SynthRadScan


from numpy.testing import assert_almost_equal

def test_runner_planar():

    willplot  = False

    Grid, Layer, Q = WAKZK_planar.WAKZK_planar()

    tol = 1E-4
    
    assert Grid.KK == 16