tests/testthat/test-Pinecone.R

In GitAI: Extracts Knowledge from 'Git' Repositories

test_that("getting index metadata", {

  db <- PineconeMocked$new(
    namespace = "test_project_id",
    index  = "gitai"
  )

  index <- db$get_index_metadata()
  index$host |> is.character() |> expect_true()
})

test_that("getting embeddings", {

  db <- PineconeMocked$new(
    namespace = "test_project_id",
    index  = "gitai"
  )

  test_text <- "Apple is a popular fruit known for its sweetness and crisp texture."
  embeddings <- db$.__enclos_env__$private$.get_embeddings(text = test_text)

  length(embeddings) |> expect_equal(1024)
})

test_that("writting records", {

  db <- PineconeMocked$new(
    namespace = "test_project_id",
    index = "gitai"
  )

  test_texts <- c(
    "Apple is a popular fruit known for its sweetness and crisp texture.",
    "The tech company Apple is known for its innovative products like the iPhone.",
    "Many people enjoy eating apples as a healthy snack.",
    "Apple Inc. has revolutionized the tech industry with its sleek designs and user-friendly interfaces.",
    "An apple a day keeps the doctor away, as the saying goes.",
    "Apple Computer Company was founded on April 1, 1976, by Steve Jobs, Steve Wozniak, and Ronald Wayne as a partnership."
  )

  for (i in seq_along(test_texts)) {

    result <- db$write_record(
      id = paste0("id_", i),
      text = test_texts[i]
    )

    result$upsertedCount |> expect_equal(1)
  }
})

test_that("finding records", {

  db <- PineconeMocked$new(
    namespace = "test_project_id",
    index = "gitai"
  )

  result <- db$find_records(
    query = "Tell me about Apple Tech computer company.",
    top_k = 1
  )

  length(result) |> expect_equal(1)
  result[[1]]$id |> expect_equal("id_2")
  result[[1]]$metadata$text |> is.character() |> expect_true()
  result[[1]]$score |> is.numeric() |> expect_true()

})

test_that("reading records", {

  db <- PineconeMocked$new(
    namespace = "test_project_id",
    index = "gitai"
  )

  result <- db$read_record(id = "id_1")

  result[[1]]$metadata$text |>
    is.character() |>
    expect_true()
})

test_that("listing all records IDs", {

  db <- PineconeMocked$new(
    namespace = "test_project_id",
    index = "gitai"
  )

  result <- db$list_record_ids()

  expect_type(result, "character")
  expect_gt(length(result), 1)
})