From bfedee10de129684cf110346f7341ba28aa05cf9 Mon Sep 17 00:00:00 2001 From: Lars Wirzenius Date: Sun, 2 Jun 2019 13:26:44 +0300 Subject: Change: move assertEqual to fable.py --- echo.py | 13 +++++-------- fable.py | 6 ++++++ 2 files changed, 11 insertions(+), 8 deletions(-) diff --git a/echo.py b/echo.py index 7a5f186..0d108f3 100644 --- a/echo.py +++ b/echo.py @@ -1,9 +1,6 @@ import subprocess -def assertEqual(a, b): - if a != b: - raise Exception( - 'expected {!r} == {!r}, but was disappointed'.format(a, b)) +import fable def _run_echo(ctx, args): cmd = '/bin/echo' @@ -23,13 +20,13 @@ def run_echo_with_args(ctx, args=None): def exit_code_is_zero(ctx, exit_code=None): exit_code = int(exit_code) - assertEqual(ctx['exit_code'], exit_code) + fable.assertEqual(ctx['exit_code'], exit_code) def stdout_is_a_newline(ctx): - assertEqual(ctx['stdout'], '\n') + fable.assertEqual(ctx['stdout'], '\n') def stdout_is_text(ctx, text=None): - assertEqual(ctx['stdout'], text + '\n') + fable.assertEqual(ctx['stdout'], text + '\n') def stderr_is_empty(ctx): - assertEqual(ctx['stderr'], '') + fable.assertEqual(ctx['stderr'], '') diff --git a/fable.py b/fable.py index 71c949a..18ac72d 100644 --- a/fable.py +++ b/fable.py @@ -30,3 +30,9 @@ class Run: def end(self): sys.stdout.write('OK: {}\n'.format(self._name)) + + +def assertEqual(a, b): + if a != b: + raise Exception( + 'expected {!r} == {!r}, but was disappointed'.format(a, b)) -- cgit v1.2.1