From ef5bbab6723f0b62082d085ac29e3f5eace80f1d Mon Sep 17 00:00:00 2001 From: Alexey Zakharenkov Date: Mon, 8 Nov 2021 12:22:57 +0300 Subject: [PATCH] Return "error_if"-logic for tram lines count --- subway_structure.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/subway_structure.py b/subway_structure.py index 280597b..b925bb5 100644 --- a/subway_structure.py +++ b/subway_structure.py @@ -1748,7 +1748,8 @@ class City: ] ) if self.found_tram_lines != self.num_tram_lines: - self.error( + log_function = self.error if self.found_tram_lines == 0 else self.notice + log_function( 'Found {} tram lines, expected {}'.format( self.found_tram_lines, self.num_tram_lines ),