def setUp(self):
self.problems = [
ManufacturedSolution("1D: no spatial", 1,
lambda t, r: t,
lambda t, k, alpha, r: k/alpha * (r*0.0+1)),
ManufacturedSolution("1D: spatial and time", 1,
lambda t, r: np.sin(t)*np.log(r),
lambda t, k, alpha, r: k/alpha * np.log(r) * np.cos(t)),
ManufacturedSolution("1D: just spatial", 1,
lambda t, r: np.sin(r),
lambda t, k, alpha, r: k * np.sin(r) - k/r*np.cos(r)),
ManufacturedSolution("2D: just time", 2,
lambda t, r, th: t,
lambda t, k, alpha, r, th: k/alpha * (r*0.0+1)),
ManufacturedSolution("2D: just r", 2,
lambda t, r, th: np.sin(r),
lambda t, k, alpha, r, th: k * np.sin(r) - k/r * np.cos(r)),
ManufacturedSolution("2D: just theta", 2,
lambda t, r, th: np.cos(th),
lambda t, k, alpha, r, th: k * np.cos(th) / r**2.0),
ManufacturedSolution("2D: theta and r", 2,
lambda t, r, th: np.cos(th) / r,
lambda t, k, alpha, r, th: -k*np.cos(th) / r**3.0),
ManufacturedSolution("2D: all three", 2,
lambda t, r, th: np.log(r) * np.sin(th) / (t+1),
lambda t, k, alpha, r, th: k*np.log(r)*np.sin(th)/((t+1)*r**2.0) - k/alpha * np.log(r) * np.sin(th) / (t+1)**2.0),
ManufacturedSolution("3D: just t", 3,
lambda t, r, th, z: t,
lambda t, k, alpha, r, th, z: k/alpha * (r*0.0+1)),
ManufacturedSolution("3D: just r", 3,
lambda t, r, th, z: np.sin(r),
lambda t, k, alpha, r, th, z: k * np.sin(r) - k/r * np.cos(r)),
ManufacturedSolution("3D: just theta", 3,
lambda t, r, th, z: np.cos(th),
lambda t, k, alpha, r, th, z: k * np.cos(th) / r**2.0),
ManufacturedSolution("3D: just z", 3,
lambda t, r, th, z: np.sin(z),
lambda t, k, alpha, r, th, z: k * np.sin(z)),
ManufacturedSolution("3D: all spatial", 3,
lambda t, r, th, z: z**2.0*np.cos(th)/r,
lambda t, k, alpha, r, th, z: -k*np.cos(th)/r*((z/r)**2.0+2)),
ManufacturedSolution("3D: everything", 3,
lambda t, r, th, z: np.log(r)*np.sin(th)*np.cos(z)/(t+1.0),
lambda t, k, alpha, r, th, z: k*np.log(r) * np.sin(th) * np.cos(z) / (t+1) * (1.0 + 1/r**2.0 - 1.0/(alpha*(t+1)))),
]
self.solver = thermal.FiniteDifferenceImplicitThermalSolver()
self.material = materials.ConstantThermalMaterial("Test", 10.0, 5.0)
self.fluid = materials.ConstantFluidMaterial({"Test": 7.5})
self.tol = 1e-2
self.atol = 1e-3
#!/usr/bin/env python3
import sys
sys.path.append('../../..')
from srlife import materials, thermal
from thermalsol import ManufacturedSolution
import numpy as np
problem = ManufacturedSolution(
"test", 1, lambda t, r: np.sin(r),
lambda t, k, alpha, r: k * np.sin(r) - k / r * np.cos(r))
def run_with(solver, material, fluid):
"""
Run the standard problems with the provided solver and material
"""
for problem in problems:
res = problem.solve(solver, material, fluid)
problem.plot_comparison(res)
plt.show()
if __name__ == "__main__":
solver = thermal.FiniteDifferenceImplicitThermalSolver()
tmat = materials.ConstantThermalMaterial("Test", 10.0, 5.0)
fmat = materials.ConstantFluidMaterial({"Test": 7.5})
if len(sys.argv) == 2:
#!/usr/bin/env python3
import sys
sys.path.append("../../..")
from srlife import materials, thermal
from thermalsol import ManufacturedSolution
import numpy as np
problem = ManufacturedSolution(
"test",
2,
lambda t, r, th: np.log(r) * np.sin(th) / (t + 1),
lambda t, k, alpha, r, th: k * np.log(r) * np.sin(th) / ((t + 1) * r**2.0)
- k / alpha * np.log(r) * np.sin(th) / (t + 1) ** 2.0,
)
def run_with(solver, material, fluid):
"""
Run the standard problems with the provided solver and material
"""
for problem in problems:
res = problem.solve(solver, material, fluid)
problem.plot_comparison(res)
plt.show()
if __name__ == "__main__":
#!/usr/bin/env python3
import sys
sys.path.append('../../..')
from srlife import materials, thermal
from thermalsol import ManufacturedSolution
import numpy as np
import matplotlib.pyplot as plt
problems = [
ManufacturedSolution("1D: no spatial", 1, lambda t, r: t,
lambda t, k, alpha, r: k / alpha * (r * 0.0 + 1)),
ManufacturedSolution(
"1D: spatial and time", 1, lambda t, r: np.sin(t) * np.log(r),
lambda t, k, alpha, r: k / alpha * np.log(r) * np.cos(t)),
ManufacturedSolution(
"1D: just spatial", 1, lambda t, r: np.sin(r),
lambda t, k, alpha, r: k * np.sin(r) - k / r * np.cos(r)),
ManufacturedSolution("2D: just time", 2, lambda t, r, th: t,
lambda t, k, alpha, r, th: k / alpha * (r * 0.0 + 1)),
ManufacturedSolution(
"2D: just r", 2, lambda t, r, th: np.sin(r),
lambda t, k, alpha, r, th: k * np.sin(r) - k / r * np.cos(r)),
ManufacturedSolution("2D: just theta", 2, lambda t, r, th: np.cos(th),
lambda t, k, alpha, r, th: k * np.cos(th) / r**2.0),
ManufacturedSolution("2D: theta and r", 2, lambda t, r, th: np.cos(th) / r,
lambda t, k, alpha, r, th: -k * np.cos(th) / r**3.0),
ManufacturedSolution(
#!/usr/bin/env python3
import sys
sys.path.append("../../..")
from srlife import materials, thermal
from thermalsol import ManufacturedSolution
import numpy as np
problem = ManufacturedSolution(
"test",
3,
lambda t, r, th, z: np.log(r) * np.sin(th) * np.cos(z) / (t + 1.0),
lambda t, k, alpha, r, th, z: k * np.log(r) * np.sin(th) * np.cos(z) /
(t + 1) * (1.0 + 1 / r**2.0 - 1.0 / (alpha * (t + 1))),
)
def run_with(solver, material, fluid):
"""
Run the standard problems with the provided solver and material
"""
for problem in problems:
res = problem.solve(solver, material, fluid)
problem.plot_comparison(res)
plt.show()
if __name__ == "__main__":