
import click.testing
import pytest

from cfgrib import __main__


def test_main():
    runner = click.testing.CliRunner()

    res = runner.invoke(__main__.cfgrib_cli, ['selfcheck'])

    assert res.exit_code == 0
    assert 'Your system is ready.' in res.output

    res = runner.invoke(__main__.cfgrib_cli, ['non-existent-command'])
    assert res.exit_code == 2


import click.testing

from cfgrib import __main__


def test_main():
    runner = click.testing.CliRunner()

    res = runner.invoke(__main__.cfgrib_cli, ['selfcheck'])

    assert res.exit_code == 0
    assert 'Your system is ready.' in res.output

    res = runner.invoke(__main__.cfgrib_cli, ['non-existent-command'])
    assert res.exit_code == 2

Fix docs and CLI tests.
Fix docs and CLI tests.

  
  import click.testing
- import pytest
  
  from cfgrib import __main__
  
  
  def test_main():
      runner = click.testing.CliRunner()
  
      res = runner.invoke(__main__.cfgrib_cli, ['selfcheck'])
  
      assert res.exit_code == 0
      assert 'Your system is ready.' in res.output
  
      res = runner.invoke(__main__.cfgrib_cli, ['non-existent-command'])
      assert res.exit_code == 2
  
Fix docs and CLI tests.
## Code Before:

import click.testing
import pytest

from cfgrib import __main__


def test_main():
    runner = click.testing.CliRunner()

    res = runner.invoke(__main__.cfgrib_cli, ['selfcheck'])

    assert res.exit_code == 0
    assert 'Your system is ready.' in res.output

    res = runner.invoke(__main__.cfgrib_cli, ['non-existent-command'])
    assert res.exit_code == 2

## Instruction:
Fix docs and CLI tests.
## Code After:

import click.testing

from cfgrib import __main__


def test_main():
    runner = click.testing.CliRunner()

    res = runner.invoke(__main__.cfgrib_cli, ['selfcheck'])

    assert res.exit_code == 0
    assert 'Your system is ready.' in res.output

    res = runner.invoke(__main__.cfgrib_cli, ['non-existent-command'])
    assert res.exit_code == 2
