import numpy as np
import math
import pdb
from sfepy import data_dir
filename_mesh = data_dir + '/meshes/2d/rectangle_fine_quad.mesh'
options = {'nls': 'newton', 'ls': 'ls'}
materials = {
#'mat_diffusion_coupling': ({'f': np.array([[0], [0]])},),
 'mat_diffusion_coupling' : 'diffusion_coupling_coefficient',
}
regions = {'Omega': ('all', {})}
fields = {'temp': ('real', 1, 'Omega', 1)}
variables = {'p': ('unknown field', 'temp', 0), 's': ('test field'
             , 'temp', 'p')}
equations = \
    {'Price': """ dw_diffusion_coupling.2.Omega(mat_diffusion_coupling.f, s, p) 
                  = 0"""}
ebcs = {}
options = {'output_format': 'vtk'}
solvers = {'ls': ('ls.scipy_direct', {}), 'newton': ('nls.newton',
           {'i_max': 1, 'eps_a': 1e-10})}

def diffusion_coupling_coefficient(ts, coors, mode=None, **kwargs):
    return {'f': np.zeros((coors.shape[0], 2, 1))}

functions = \
    {'diffusion_coupling_coefficient': (diffusion_coupling_coefficient,
     )}