Add test for `NA` values