Coverage for tcvx21/__init__.py: 100%

1# This file helps to simplify import paths -- so you can type `import tcvx21.read_from_json` instead of 

2# `from tcvx21.json_io_m import read_from_json` 

3# Only key functionality is imported here -- use the full path to access non-key functionality 

4 

5from pathlib import Path 

6repository_dir = Path(__file__).parents[1] 

7library_dir = Path(__file__).parent 

8sample_data = repository_dir / 'tests' / 'sample_data' 

9 

10experimental_reference_dir = repository_dir / '1.experimental_data' 

11divertor_coords_json = experimental_reference_dir / 'reference_scenario/divertor_polygon.json' 

12rdpa_coords_json = experimental_reference_dir / 'reference_scenario/RDPA_reference_coords.json' 

13thomson_coords_json = experimental_reference_dir / 'reference_scenario/thomson_position.json' 

14 

15simulation_dir = repository_dir / '2.simulation_data' 

16grillix_dir = simulation_dir / 'GRILLIX_2021' 

17grillix_forward_dir = grillix_dir / 'checkpoints_for_1mm/forward_field' 

18grillix_reversed_dir = grillix_dir / 'checkpoints_for_1mm/reversed_field' 

19gbs_dir = simulation_dir / 'GBS_2021' 

20tokam3x_dir = simulation_dir / 'TOKAM3X_2021' 

21 

22notebooks_dir = repository_dir / 'notebooks' 

23test_dir = repository_dir / 'tests' 

24results_dir = repository_dir / '3.results' 

25 

26test_figures_dir = results_dir / 'test_fig' 

27 

28for directory in [repository_dir, library_dir, 

29 experimental_reference_dir, simulation_dir, notebooks_dir, test_dir]: 

30 assert directory.exists() and directory.is_dir(), f"{directory} is not a directory" 

31 

32style_sheet = library_dir / 'tcvx21.mplstyle' 

33style_sheet_inline = library_dir / 'tcvx21_inline.mplstyle' 

34assert style_sheet.exists() 

35 

36from .units_m import Quantity, Dimensionless, convert_xarray_to_quantity, unit_registry 

37from .file_io import read_from_json, write_to_json, read_struct_from_file 

38from .record_c import Record, EmptyRecord, template_file 

39from .dict_methods_m import summarise_tree_dict, recursive_rename_keys, recursive_assert_dictionaries_equal 

40 

41 

42# Add subdirectories to namespace (can be called as tcvx21.grillix_post) 

43from . import grillix_post 

44from . import analysis 

45from . import file_io 

46from . import observable_c