code: def test_sqrt_neg_inf() -> None:
      out = pl.DataFrame(
          {
              "val": [float("-Inf"), -9, 0, 9, float("Inf")],
          }
      ).with_columns(pl.col("val").sqrt().alias("sqrt"))
      # comparing nans and infinities by string value as they are not cmp
      assert str(out["sqrt"].to_list()) == str(
          [float("nan"), float("nan"), 0.0, 3.0, float("Inf")]
      )
  
file_path: polars/py-polars/tests/unit/operations/test_arithmetic.py
imports: from datetime import date, datetime, timedelta
  import numpy
  import pytest
  import polars
  from polars.testing import assert_series_equal
label_desc: Use image node_id 1 for calling a global function with example usage: test_sqrt_neg_inf() without return types