@@ -104,14 +104,30 @@ def save_vtu_from_dict(
104104 value_keys : Tuple [str , ...],
105105 num_timestamps : int = 1 ,
106106):
107- """Save dict data to '*.vtu' file.
107+ """
108+ Save dict data to '*.vtu' file.
108109
109110 Args:
110111 filename (str): Output filename.
111112 data_dict (Dict[str, np.ndarray]): Data in dict.
112113 coord_keys (Tuple[str, ...]): Tuple of coord key. such as ("x", "y").
113114 value_keys (Tuple[str, ...]): Tuple of value key. such as ("u", "v").
114115 num_timestamps (int, optional): Number of timestamp in data_dict. Defaults to 1.
116+
117+ Examples:
118+ >>> import ppsci
119+ >>> import numpy as np
120+ >>> filename = "path/to/file.vtu"
121+ >>> data_dict = {
122+ ... "x": np.array([[1], [2], [3],[4]]),
123+ ... "y": np.array([[2], [3], [4],[4]]),
124+ ... "z": np.array([[3], [4], [5],[4]]),
125+ ... "u": np.array([[4], [5], [6],[4]]),
126+ ... "v": np.array([[5], [6], [7],[4]]),
127+ ... }
128+ >>> coord_keys = ("x","y","z")
129+ >>> value_keys = ("u","v")
130+ >>> ppsci.visualize.save_vtu_from_dict(filename, data_dict, coord_keys, value_keys) # doctest: +SKIP
115131 """
116132 if len (coord_keys ) not in [2 , 3 , 4 ]:
117133 raise ValueError (f"ndim of coord ({ len (coord_keys )} ) should be 2, 3 or 4" )
0 commit comments