Skip to content

Commit 559b37e

Browse files
committed
fix repr of Model with no process
1 parent d345c94 commit 559b37e

2 files changed

Lines changed: 8 additions & 2 deletions

File tree

xsimlab/model.py

Lines changed: 5 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -355,9 +355,12 @@ def drop_processes(self, keys):
355355
def __repr__(self):
356356
n_inputs = sum([len(v) for v in self._input_vars.values()])
357357

358-
hdr = ("<xsimlab.Model (%d processes, %d inputs)>\n"
358+
hdr = ("<xsimlab.Model (%d processes, %d inputs)>"
359359
% (len(self._processes), n_inputs))
360360

361+
if not len(self._processes):
362+
return hdr
363+
361364
max_line_length = 70
362365
col_width = max([_calculate_col_width(var)
363366
for var in self._input_vars.values()])
@@ -378,4 +381,4 @@ def __repr__(self):
378381
proc_str += '\n' + '\n'.join(lines)
379382
blocks.append(proc_str)
380383

381-
return hdr + '\n'.join(blocks)
384+
return hdr + '\n' + '\n'.join(blocks)

xsimlab/tests/test_model.py

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -157,3 +157,6 @@ def test_drop_processes(self, model):
157157

158158
def test_repr(self, model, model_repr):
159159
assert repr(model) == model_repr
160+
161+
expected = "<xsimlab.Model (0 processes, 0 inputs)>"
162+
assert repr(Model({})) == expected

0 commit comments

Comments
 (0)