Skip to content

Commit aa9e2bd

Browse files
authored
Add datatree repr asv (#10214)
1 parent dd446d7 commit aa9e2bd

File tree

1 file changed

+19
-0
lines changed

1 file changed

+19
-0
lines changed

Diff for: asv_bench/benchmarks/repr.py

+19
Original file line numberDiff line numberDiff line change
@@ -25,6 +25,25 @@ def time_repr_html(self):
2525
self.ds._repr_html_()
2626

2727

28+
class ReprDataTree:
29+
def setup(self):
30+
# construct a datatree with 500 nodes
31+
number_of_files = 20
32+
number_of_groups = 25
33+
tree_dict = {}
34+
for f in range(number_of_files):
35+
for g in range(number_of_groups):
36+
tree_dict[f"file_{f}/group_{g}"] = xr.Dataset({"g": f * g})
37+
38+
self.dt = xr.DataTree.from_dict(tree_dict)
39+
40+
def time_repr(self):
41+
repr(self.dt)
42+
43+
def time_repr_html(self):
44+
self.dt._repr_html_()
45+
46+
2847
class ReprMultiIndex:
2948
def setup(self):
3049
index = pd.MultiIndex.from_product(

0 commit comments

Comments
 (0)