Coverage for pySDC / projects / parallelSDC_reloaded / kaps_setup.py: 100%

1#!/usr/bin/env python3 

2# -*- coding: utf-8 -*- 

3""" 

4Created on Thu Dec 7 19:29:46 2023 

5 

6Setup script for the Kaps problem 

7""" 

8 

9import numpy as np 

10import matplotlib.pyplot as plt 

11 

12from pySDC.projects.parallelSDC_reloaded.utils import solutionExact, getParamsRK, solutionSDC 

13 

14script = __file__.split('/')[-1].split('.')[0] 

15 

16tEnd = 1 

17nSteps = 100 

18epsilon = 1e-3 

19rkScheme = "DIRK43" 

20 

21tVals = np.linspace(0, tEnd, nSteps + 1) 

22 

23print("Computing ODE solution") 

24uExact = solutionExact(tEnd, nSteps, "KAPS", epsilon=epsilon) 

25 

26params = getParamsRK(rkScheme) 

27uNum, counters, parallel = solutionSDC(tEnd, nSteps, params, 'KAPS', epsilon=epsilon) 

28 

29figName = f"{script}_solution" 

30plt.figure(figName) 

31plt.plot(tVals, uExact[:, 0], '-', label="x-exact") 

32plt.plot(tVals, uExact[:, 1], '-', label="y-exact") 

33plt.plot(tVals, uNum[:, 0], '--', label="x-num") 

34plt.plot(tVals, uNum[:, 1], '--', label="y-num") 

35 

36plt.legend() 

37plt.xlabel("time") 

38plt.ylabel("solution") 

39plt.tight_layout()