code: def test_cli(monkeypatch):
      query = "What is the Earth?"
      monkeypatch.setattr("builtins.input", lambda _: query)
  
      from src.gen import main
  
      all_generations = main(
          base_model="gptj",
          cli=True,
          cli_loop=False,
          score_model="None",
      )
  
      assert len(all_generations) == 1
      assert (
          "The Earth is a planet in our solar system"
          in all_generations[0]
      )
  
file_path: h2ogpt/tests/test_cli.py
imports: import pytest
  from tests.utils import wrap_test_forked, get_llama
  from src.enums import DocumentSubset
label_desc: Use image node_id 1 for calling a global function with example usage: test_cli(monkeypatch) without return types