fix nimparser unittests

Signed-off-by: Varun Ramachandra Sekar <vsekar@nvidia.com>

Author: varunrsekar

…ternal/nimparser/nimparser_suite_test.go …nal/nimparser/v1/nimparser_suite_test.gointernal/nimparser/nimparser_suite_test.go renamed to internal/nimparser/v1/nimparser_suite_test.go internal/nimparser/nimparser_suite_test.go renamed to internal/nimparser/v1/nimparser_suite_test.go

@@ -14,7 +14,7 @@ See the License for the specific language governing permissions and
 limitations under the License.
 */
 
-package nimparser_test
+package v1_test
 
 import (
 	"testing"
@@ -23,7 +23,7 @@ import (
 	. "github.com/onsi/gomega"
 )
 
-func TestRender(t *testing.T) {
+func TestNIMParser(t *testing.T) {
 	RegisterFailHandler(Fail)
 	RunSpecs(t, "Model Manifest Parser Test Suite")
 }

internal/nimparser/v1/nimparser_test.go

@@ -32,7 +32,7 @@ var _ = Describe("NIMParser", func() {
 	Context("ParseModelManifest", func() {
 		It("should parse a model profile for trtllm engine files correctly", func() {
 
-			filePath := filepath.Join("testdata", "manifest_trtllm.yaml")
+			filePath := filepath.Join("testdata", "manifest_trtllm_v1.yaml")
 			nimparser := NIMParser{}
 			config, err := nimparser.ParseModelManifest(filePath)
 			Expect(err).NotTo(HaveOccurred())
@@ -49,7 +49,7 @@ var _ = Describe("NIMParser", func() {
 		})
 		It("should parse a model profile for vllm engine files correctly", func() {
 
-			filePath := filepath.Join("testdata", "manifest_vllm.yaml")
+			filePath := filepath.Join("testdata", "manifest_vllm_v1.yaml")
 			nimparser := NIMParser{}
 			config, err := nimparser.ParseModelManifest(filePath)
 			Expect(err).NotTo(HaveOccurred())
@@ -83,7 +83,7 @@ var _ = Describe("NIMParser", func() {
 			Expect(profile.ContainerURL).To(Equal("nvcr.io/nim/meta/llama3-70b-instruct:1.0.0"))
 		})
 		It("should match model profiles with valid parameters", func() {
-			filePath := filepath.Join("testdata", "manifest_trtllm.yaml")
+			filePath := filepath.Join("testdata", "manifest_trtllm_v1.yaml")
 			nimparser := NIMParser{}
 			config, err := nimparser.ParseModelManifest(filePath)
 			Expect(err).NotTo(HaveOccurred())
@@ -112,7 +112,7 @@ var _ = Describe("NIMParser", func() {
 			Expect(matchedProfiles).To(HaveLen(1))
 		})
 		It("should not match model profiles with invalid parameters", func() {
-			filePath := filepath.Join("testdata", "manifest_trtllm.yaml")
+			filePath := filepath.Join("testdata", "manifest_trtllm_v1.yaml")
 			nimparser := NIMParser{}
 			config, err := nimparser.ParseModelManifest(filePath)
 			Expect(err).NotTo(HaveOccurred())
@@ -133,7 +133,7 @@ var _ = Describe("NIMParser", func() {
 			Expect(matchedProfiles).To(BeEmpty())
 		})
 		It("should match model profiles using automatically discovered GPUs", func() {
-			filePath := filepath.Join("testdata", "manifest_trtllm.yaml")
+			filePath := filepath.Join("testdata", "manifest_trtllm_v1.yaml")
 			nimparser := NIMParser{}
 			config, err := nimparser.ParseModelManifest(filePath)
 			Expect(err).NotTo(HaveOccurred())

…al/nimparser/testdata/manifest_lora.yaml …nimparser/v1/testdata/manifest_lora.yamlinternal/nimparser/testdata/manifest_lora.yaml renamed to internal/nimparser/v1/testdata/manifest_lora.yaml internal/nimparser/testdata/manifest_lora.yaml renamed to internal/nimparser/v1/testdata/manifest_lora.yaml

@@ -11,3 +11,11 @@
     profile: throughput
     tp: '8'
   container_url: nvcr.io/nim/meta/llama3-70b-instruct:1.0.0
+  workspace: !workspace
+    components:
+    - dst: trtllm_engine
+      src:
+        files:
+          - !name 'LICENSE.txt'
+          - !name 'checksums.blake3'
+          - !name 'config.json'

…nimparser/testdata/manifest_non_llm.yaml …parser/v1/testdata/manifest_non_llm.yamlinternal/nimparser/testdata/manifest_non_llm.yaml renamed to internal/nimparser/v1/testdata/manifest_non_llm.yaml internal/nimparser/testdata/manifest_non_llm.yaml renamed to internal/nimparser/v1/testdata/manifest_non_llm.yaml

@@ -8,3 +8,11 @@
     precision: fp16
     tp: '8'
   container_url: nvcr.io/nim/meta/llama3-70b-instruct:1.0.0
+  workspace: !workspace
+    components:
+      - dst: ''
+        src:
+          files:
+            - !name 'README.md'
+            - !name 'checksums.blake3'
+            - !name 'config.json'