1library(pystr)
2context("pystr_capitalize")
3
4test_that("it returns itself if it's already capitalized", {
5  original = "Hello"
6  capitalized = pystr_capitalize(original)
7  expect_equal(capitalized, "Hello")
8})
9
10test_that("it capitalizes the first letter only", {
11  original = "hello"
12  capitalized = pystr_capitalize(original)
13  expect_equal(capitalized, "Hello")
14})
15
16test_that("it decapitalizes letters beyond the first letter", {
17  original = "hello WORLD"
18  capitalized = pystr_capitalize(original)
19  expect_equal(capitalized, "Hello world")
20})
21
22test_that("it works with vectors of strings", {
23  original = c("HELLO", "WORLD")
24  capitalized = pystr_capitalize(original)
25  expect_equal(capitalized, c("Hello", "World"))
26})
27
28test_that("it handles NAs appropriately", {
29  original = c("HELLO", NA)
30  capitalized = pystr_capitalize(original)
31  expect_equal(capitalized, c("Hello", NA))
32})
33