code: def test_location_args():
      """Test some data types for a location arg."""
      location = np.array([45.5236, -122.6750])
      m = folium.Map(location)
      assert m.location == [45.5236, -122.6750]
  
      df = pd.DataFrame({"location": [45.5236, -122.6750]})
      m = folium.Map(df["location"])
      assert m.location == [45.5236, -122.6750]
  
imports: import json
  import os
  import branca.element
  import geopandas
  import jinja2
  import numpy
  import pandas
  import pytest
  import xyzservices.providers
  from jinja2 import Environment, PackageLoader
  from jinja2.utils import htmlsafe_json_dumps
  import folium
  from folium import TileLayer
  from folium.features import Choropleth, GeoJson
label_desc: Use image node_id 3 for calling a global function with example usage: test_location_args() without return types