package ai

import "testing"

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

func TestInferCloudVenueClassAirportSSID(t *testing.T) {
	if InferCloudVenueClass(CloudVenueReport{SSID: "JFK-Free-WiFi"}) != CloudVenueAirport {
		t.Fatal("expected airport class")
	}
}