package ai

import "testing"

func TestInferCloudVenueClassGPU(t *testing.T) {
	if InferCloudVenueClass(CloudVenueReport{InstanceType: "g4dn.xlarge"}) != CloudVenueGPU {
		t.Fatal("expected gpu class")
	}
}

func TestInferCloudVenueClassInteractive(t *testing.T) {
	if InferCloudVenueClass(CloudVenueReport{InstanceType: "t3.micro"}) != CloudVenueInteractive {
		t.Fatal("expected interactive class")
	}
}