code: def test_squared_difference(
      data: np.ndarray, params: dict, expected: Union[int, ValueError]
  ):
      try:
          out = squared_difference(a=data["a"], b=data["b"])
          assert out == expected, "Test failed."
      except ValueError:
          pass
  
imports: import pytest
  from pytest_lazyfixture import lazy_fixture
  from quantus.functions.loss_func import mse
  from quantus.functions.similarity_func import *
label_desc: Use image node_id 25 for calling a global function with example usage: test_squared_difference(data, params, expected) without return types