code: def test_constant_cost_per_treatment_feature(self, cost_manager):
      with patch.object(
          cost_manager, "_create_policy", return_value=None
      ) as mock_create:
          try:
              cost_manager.add(
                  ["AveRooms", "Population", "AveOccup"],
                  treatment_cost=[1, 2, 3],
              )
              cost_manager.compute()
          except TypeError:
              pass
          mock_create.assert_any_call(
              ANY, ANY, "AveRooms", 1, ANY, ANY, ANY
          )
          mock_create.assert_any_call(
              ANY, ANY, "Population", 2, ANY, ANY, ANY
          )
          mock_create.assert_any_call(
              ANY, ANY, "AveOccup", 3, ANY, ANY, ANY
          )
  
imports: import os
  from unittest.mock import ANY, patch
  import numpy
  import pytest
  from raiutils.exceptions import UserConfigValidationException
  from raiutils.models import ModelTask
  from responsibleai import RAIInsights
  from responsibleai._interfaces import CausalData
  from responsibleai._internal.constants import FileFormats
  from responsibleai._tools.shared.state_directory_management import DirectoryManager
  from responsibleai.managers.causal_manager import CausalManager
label_desc: Use image node_id 5 for calling the TestCausalManagerTreatmentCosts obj's underlying member method code with example usage: obj.test_constant_cost_per_treatment_feature(cost_manager) without return types