code: def test_output():
      """
      Test that you can get an expected output shape from the TTS
      """
      ts = TransformerTreeStack(3, 5, 0.0)
      initial = ts.initial_state()
      out = ts.output(initial)
      assert out.shape == torch.Size([5])
      assert torch.allclose(initial.value.output, out)
  
file_path: stanza/stanza/tests/constituency/test_transformer_tree_stack.py
imports: import pytest
  import torch
  from stanza.models.constituency.transformer_tree_stack import TransformerTreeStack
label_desc: Use image node_id 2 for calling a global function with example usage: test_output() without return types