[NVIDIA-IMEX] Test

Author: Himani Anil Deshpande

cookbooks/aws-parallelcluster-platform/resources/nvidia_imex/partial/_nvidia_imex_common.rb

@@ -54,6 +54,7 @@
   end
   # Save Imex version in Node Attributes for InSpec Tests
   node.default['cluster']['nvidia']['imex']['version'] = _nvidia_imex_version
+  node.default['cluster']['nvidia']['imex']['package'] = "#{nvidia_imex_service}-#{node['cluster']['nvidia']['driver_version'].split('.')[0]}"
   node_attributes 'dump node attributes'
 end
 

cookbooks/aws-parallelcluster-platform/test/controls/nvidia_imex_spec.rb

@@ -14,7 +14,7 @@
 
   nvidia_imex_service = 'nvidia-imex'
 
-  describe package(nvidia_imex_service) do
+  describe package("#{node['cluster']['nvidia']['imex']['package']}") do
     it { should be_installed }
     its('version') { should match /#{node['cluster']['nvidia']['imex']['version']}/ }
   end