def test_output_in_right_format(self, memory_instance: str, scene_type: str):
    """Tests that the component extracts the scene type correctly.

    Args:
      memory_instance: The string that the associative memory returns.
      scene_type: The expected scene type.
    """
    memory = mock.create_autospec(associative_memory.AssociativeMemory,
                                  instance=True)
    memory.retrieve_by_regex.return_value = [memory_instance]
    component = current_scene.CurrentScene(name="current scene",
                                           memory=memory)
    self.assertEqual(component.state(), "")
    component.update()
    self.assertEqual(component.state(), scene_type)
Tests that the component extracts the scene type correctly.

    Args:
      memory_instance: The string that the associative memory returns.
      scene_type: The expected scene type.
    
concordia/components/game_master/deprecated/current_scene_test.py