old_contents: import pytest
  
  from pelops.features.keras_model import KerasModelFeatureProducer
  
  
  def test_load_model_workaround():
      # @TODO get some environment variable set when in CI environment
      # test to see, modify path...
      model_filename = '/home/ubuntu/pelops/testci/small.json'
      weight_filename = '/home/ubuntu/pelops/testci/small.hdf5'
      model = KerasModelFeatureProducer.load_model_workaround(
          model_filename, weight_filename)
      assert model.layers[0].name == 'dense_1'
  
new_contents: import pytest
  
  from pelops.features.keras_model import KerasModelFeatureProducer
  
  
  def test_load_model_workaround():
      # @TODO get some environment variable set when in CI environment
      # test to see, modify path...
      model_filename = '/home/ubuntu/pelops/testci/small.json'
      weight_filename = '/home/ubuntu/pelops/testci/small.hdf5'
      model = KerasModelFeatureProducer.load_model_workaround(
          model_filename, weight_filename)
      assert model.layers[0].name == 'dense_8'
  
subject: correct value