code: def test_input_shapes(
      self, temperature, humidity, distance, sample_rate
  ):
      np.random.seed(1)
  
      samples = get_chirp_test(sample_rate, 10)
  
      augment = AirAbsorption(
          min_temperature=temperature,
          max_temperature=temperature,
          min_humidity=humidity,
          max_humidity=humidity,
          min_distance=distance,
          max_distance=distance,
      )
  
      # Test 1D case
      processed_samples = augment(samples, sample_rate=sample_rate)
      assert processed_samples.shape == samples.shape
      assert processed_samples.dtype == np.float32
  
      # Test 2D case
      samples = np.tile(samples, (2, 1))
      processed_samples = augment(samples, sample_rate=sample_rate)
      assert processed_samples.shape == samples.shape
      assert processed_samples.dtype == np.float32
  
file_path: audiomentations/tests/test_air_absorption.py
imports: from audiomentations import AirAbsorption
  import numpy
  import pytest
  import scipy
  import scipy.signal
label_desc: Use image node_id 1 for calling the TestAirAbsorptionTransform obj's underlying member method code with example usage: obj.test_input_shapes(temperature, humidity, distance, sample_rate) without return types