test_diff.py

# Without the improvements
test_something():
    try:
        entity = make_entity()
    except CLIFactoryError as err:
        self.fail(err)
        
    result = Entity.list()
    assert result.return_code == 0
    assert len(result.stderr) == 0
    assert result.stdout['somedata'] == 'expected_data'
    

# With the improvements
test_something():
    entity = make_entity()
    result = Entity.list()
    assert result.stdout['somedata'] == 'expected_data'