code: def test_generate_continuation(self):
      ag = self.get_audiogen()
      prompt = torch.randn(3, 1, 16000)
      wav = ag.generate_continuation(prompt, 16000)
      assert list(wav.shape) == [3, 1, 32000]
  
      prompt = torch.randn(2, 1, 16000)
      wav = ag.generate_continuation(
          prompt, 16000, ["youpi", "lapin dort"]
      )
      assert list(wav.shape) == [2, 1, 32000]
  
      prompt = torch.randn(2, 1, 16000)
      with pytest.raises(AssertionError):
          wav = ag.generate_continuation(
              prompt, 16000, ["youpi", "lapin dort", "one too many"]
          )
  
file_path: audiocraft/tests/models/test_audiogen.py
imports: import pytest
  import torch
  from audiocraft.models import AudioGen
label_desc: Use image node_id 3 for calling the TestAudioGenModel obj's underlying member method code with example usage: obj.test_generate_continuation() without return types