code: def test_set_locations_np_array(self):
      import numpy as np
  
      heatmap = Heatmap(locations=self.locations)
      heatmap.locations = np.array(self.locations * 2)
      assert heatmap.locations == self.locations * 2
  
imports: import unittest
  import pytest
  import traitlets
  from ..heatmap import _HeatmapOptionsMixin, heatmap_layer, Heatmap, WeightedHeatmap
  from ..geotraitlets import InvalidPointException, InvalidWeightException
label_desc: Use image node_id 2 for calling the TestHeatmap obj's underlying member method code with example usage: obj.test_set_locations_np_array() without return types