mirror of https://github.com/mingrammer/diagrams
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
30 lines
980 B
30 lines
980 B
#!/usr/bin/env python3
|
|
|
|
"""
|
|
Test script for new AWS AI/ML service icons
|
|
Tests Amazon Nova, AWS App Studio, Amazon CodeWhisperer, and AWS Neuron
|
|
"""
|
|
|
|
from diagrams import Diagram
|
|
from diagrams.aws.ml import AmazonNova, AWSAppStudio, AmazonCodewhisperer, AWSNeuron
|
|
|
|
def test_nova_icons():
|
|
"""Test the new Nova and AI service icons"""
|
|
|
|
with Diagram("New AWS AI/ML Services", show=False, filename="nova_test"):
|
|
# Create instances of the new services
|
|
nova = AmazonNova("Amazon Nova")
|
|
app_studio = AWSAppStudio("AWS App Studio")
|
|
codewhisperer = AmazonCodewhisperer("CodeWhisperer")
|
|
neuron = AWSNeuron("AWS Neuron")
|
|
|
|
# Create a simple flow
|
|
nova >> app_studio >> codewhisperer >> neuron
|
|
|
|
print("✅ Test diagram created successfully!")
|
|
print("📁 Generated file: nova_test.png")
|
|
print("🔍 Please verify the icons render correctly in the diagram")
|
|
|
|
if __name__ == "__main__":
|
|
test_nova_icons()
|