diff options
Diffstat (limited to 'internal/image/styles_test.go')
| -rw-r--r-- | internal/image/styles_test.go | 27 |
1 files changed, 27 insertions, 0 deletions
diff --git a/internal/image/styles_test.go b/internal/image/styles_test.go new file mode 100644 index 0000000..8eeb7c1 --- /dev/null +++ b/internal/image/styles_test.go @@ -0,0 +1,27 @@ +package image + +import "testing" + +func TestArtisticStyles(t *testing.T) { + if len(ArtisticStyles) == 0 { + t.Fatal("ArtisticStyles should not be empty") + } + + if !hasStyle(ArtisticStyles, "Photorealism") { + t.Fatal(`ArtisticStyles should include "Photorealism"`) + } + + if !hasStyle(ArtisticStyles, "Candid Photography") { + t.Fatal(`ArtisticStyles should include "Candid Photography"`) + } +} + +func hasStyle(styles []string, want string) bool { + for _, style := range styles { + if style == want { + return true + } + } + + return false +} |
