tests/testthat/test_images.R

In openaistream: Streaming Capabilities for 'OpenAI API' Interactions

test_that("images",{
  skip_on_cran()
  library(png)
  Sys.setenv(TEST_EX_COND = "")
  handle_openai<-openai$new(Sys.getenv("OPENAI_KEY"))
  if(Sys.getenv("USE_PROXY")=="TRUE"){
    handle_openai$set_proxy("10.0.108.36",7890)
  }
  pic1<-handle_openai$images$create(prompt = "A small bird flies over the ocean, heading towards the storm.")
  test_dir = tempdir()
  expect_contains(object = names(pic1),expected = "data")
  pic_file1<-paste0(test_dir,"/","pic1.png")
  req <- request(pic1$data$url) %>%req_perform(path = pic_file1)
  expect_equal(req$status_code,200)
  expect_true(file.exists(pic_file1))
  #Here is the image test.
  #For simple test,you need install 'png' package.
  #Note that when a mask is not provided, OpenAI defaults to using areas of the uploaded image with a transparency of 0 as the modification area.
  #The following test adds full image transparency of 0 (modifying the entire image) to pic1.
  pic1<-readPNG(pic_file1)
  expect_type(pic1,"double")
  expect_equal(dim(pic1)[3],3)

  alpha_channel <- matrix(0, nrow = dim(pic1)[1], ncol = dim(pic1)[2])
  pic1_b<-array(c(pic1,alpha_channel), c(1024,1024,4))
  expect_equal(dim(pic1_b)[3],4)
  pic_file2<-paste0(test_dir,"/","pic2.png")
  writePNG(pic1_b,target = pic_file2)
  expect_true(file.exists(pic_file2))
  pic3<-handle_openai$images$edit(image = pic_file2,prompt = "Please modify the bird in the picture to a flock of birds."
                         ,verbosity = 3,n="2",size="256x256")
  expect_contains(object = names(pic3),expected = "data")
  expect_equal(length(pic3$data$url),2)
  pic_file3_1<-paste0(test_dir,"/","pic3_1.png")
  pic_file3_2<-paste0(test_dir,"/","pic3_2.png")
  request(pic3$data$url[1]) %>% req_perform(path = pic_file3_1)
  request(pic3$data$url[2]) %>% req_perform(path = pic_file3_2)
  expect_true(file.exists(pic_file3_1))
  expect_true(file.exists(pic_file3_2))

  pic4<-handle_openai$images$variation(image = pic_file1,n="2",size="256x256")
  expect_contains(object = names(pic4),expected = "data")
  expect_equal(length(pic4$data$url),2)
  pic_file4_1<-paste0(test_dir,"/","pic4_1.png")
  pic_file4_2<-paste0(test_dir,"/","pic4_2.png")
  request(pic4$data$url[1]) %>%req_perform(path = pic_file4_1)
  request(pic4$data$url[2]) %>%req_perform(path = pic_file4_2)
  expect_true(file.exists(pic_file4_1))
  expect_true(file.exists(pic_file4_2))

  ##error test
  pic1<-handle_openai$images$create(prompt = "A small bird flies over the ocean, heading towards the storm.",verbosity = 4)
  expect_true(!pic1$success)

  pic3<-handle_openai$images$edit(image = NULL,prompt = "Please modify the bird in the picture to a flock of birds."
                                   ,n="2",size="256x256")
  expect_true(!pic3$success)
  pic3<-handle_openai$images$edit(image = pic_file2,mask="/tmp/sadasdasd",prompt = "Please modify the bird in the picture to a flock of birds."
                                   ,n="2",size="256x256")
  expect_true(!pic3$success)
  pic3<-handle_openai$images$edit(image = pic_file2,mask=pic_file2,prompt = "Please modify the bird in the picture to a flock of birds."
                                   ,n="2",size="256x256")
  expect_contains(object = names(pic3),expected = "data")
  expect_equal(length(pic3$data$url),2)
  pic3<-handle_openai$images$edit(image = pic_file2,mask=pic_file1,prompt = "Please modify the bird in the picture to a flock of birds."
                                   ,n="2",size="256x256")
  expect_true(!pic3$success)
  pic4<-handle_openai$images$variation(image = NULL,n="2",size="256x256")
  expect_true(!pic4$success)
  pic4<-handle_openai$images$variation(image = pic_file1,n="2",size="256x256",verbosity = 4)
  expect_true(!pic4$success)
})