Update test suite to support vectorization on complex-only functions

Author: WardBrian

test/unit/math/mix/fun/acos_test.cpp

@@ -12,7 +12,7 @@ TEST(mathMixMatFun, acos) {
   // can't autodiff acos through integers
   for (auto x : stan::test::internal::common_nonzero_args())
     stan::test::expect_unary_vectorized(f, x);
-  expect_unary_vectorized<stan::test::PromoteToComplex::No>(
+  expect_unary_vectorized<stan::test::ScalarSupport::Real>(
       f, -2.2, -0.8, 0.5, 1 + std::numeric_limits<double>::epsilon(), 1.5, 3,
       3.4, 4);
   for (double re : std::vector<double>{-0.2, 0, 0.3}) {

test/unit/math/mix/fun/acosh_test.cpp

@@ -8,8 +8,8 @@ TEST(mathMixMatFun, acosh) {
   };
   for (double x : stan::test::internal::common_args())
     stan::test::expect_unary_vectorized(f, x);
-  stan::test::expect_unary_vectorized<stan::test::PromoteToComplex::Yes>(
-      f, 1.5, 3.2, 5, 10, 12.9);
+  stan::test::expect_unary_vectorized<
+      stan::test::ScalarSupport::RealAndComplex>(f, 1.5, 3.2, 5, 10, 12.9);
   // avoid pole at complex zero that can't be autodiffed
   for (double re : std::vector<double>{-0.2, 0, 0.3}) {
     for (double im : std::vector<double>{-0.3, 0.2}) {

test/unit/math/mix/fun/atan_test.cpp

@@ -8,7 +8,7 @@ TEST(mathMixMatFun, atan) {
     return atan(x);
   };
   stan::test::expect_common_nonzero_unary_vectorized<
-      stan::test::PromoteToComplex::No>(f);
+      stan::test::ScalarSupport::Real>(f);
   stan::test::expect_unary_vectorized(f, -2.6, -2, -0.2, 0.5, 1, 1.3, 1.5, 3);
   // avoid 0 imaginary component where autodiff doesn't work
   for (double re : std::vector<double>{-0.2, 0, 0.3}) {

test/unit/math/mix/fun/cbrt_test.cpp

@@ -3,7 +3,7 @@
 TEST(mathMixMatFun, cbrt) {
   auto f = [](const auto& x1) { return stan::math::cbrt(x1); };
   stan::test::expect_common_nonzero_unary_vectorized<
-      stan::test::PromoteToComplex::No>(f);
+      stan::test::ScalarSupport::Real>(f);
   stan::test::expect_unary_vectorized(f, -2.6, -2, 1, 1.3, 3);
 }
 

test/unit/math/mix/fun/conj_test.cpp

@@ -3,23 +3,11 @@
 #include <complex>
 #include <vector>
 
-namespace stan {
-namespace math {
-/**
- *  Dummy overload to allow expect_unary_vectorized. It requires real-valued
- * overloads.
- */
-template <typename V, require_not_st_complex<V>* = nullptr>
-inline auto conj(const V& z) {
-  return z;
-}
-}  // namespace math
-}  // namespace stan
-
 TEST(mathMixMatFun, conj) {
   auto f = [](const auto& x) { return stan::math::conj(x); };
   stan::test::expect_complex_common(f);
-  stan::test::expect_unary_vectorized<stan::test::PromoteToComplex::Yes>(f);
+  stan::test::expect_unary_vectorized<stan::test::ScalarSupport::ComplexOnly>(
+      f);
 }
 
 template <typename T>

test/unit/math/mix/fun/cos_test.cpp

@@ -6,8 +6,9 @@ TEST(mathMixMatFun, cos) {
     return cos(x);
   };
   stan::test::expect_common_nonzero_unary_vectorized(f);
-  stan::test::expect_unary_vectorized<stan::test::PromoteToComplex::Yes>(
-      f, -2.6, -2, -0.2, -0.5, 0, 1.5, 3, 5, 5.3);
+  stan::test::expect_unary_vectorized<
+      stan::test::ScalarSupport::RealAndComplex>(f, -2.6, -2, -0.2, -0.5, 0,
+                                                 1.5, 3, 5, 5.3);
   stan::test::expect_complex_common(f);
 }
 

test/unit/math/mix/fun/cosh_test.cpp

@@ -6,8 +6,9 @@ TEST(mathMixMatFun, cosh) {
     return cosh(x1);
   };
   stan::test::expect_common_unary_vectorized(f);
-  stan::test::expect_unary_vectorized<stan::test::PromoteToComplex::Yes>(
-      f, -2.6, -2, -1.2, -0.2, 0.5, 1, 1.3, 1.5);
+  stan::test::expect_unary_vectorized<
+      stan::test::ScalarSupport::RealAndComplex>(f, -2.6, -2, -1.2, -0.2, 0.5,
+                                                 1, 1.3, 1.5);
   stan::test::expect_complex_common(f);
 }
 

test/unit/math/mix/fun/digamma_test.cpp

@@ -3,7 +3,7 @@
 TEST(mathMixMatFun, digamma) {
   auto f = [](const auto& x1) { return stan::math::digamma(x1); };
   stan::test::expect_common_nonzero_unary_vectorized<
-      stan::test::PromoteToComplex::No>(f);
+      stan::test::ScalarSupport::Real>(f);
   stan::test::expect_unary_vectorized(f, -25, -10.2, -1.2, -1, 2.3, 5.7);
 }
 

test/unit/math/mix/fun/exp_test.cpp

@@ -5,10 +5,11 @@ TEST(mathMixMatFun, exp) {
     using stan::math::exp;
     return exp(x);
   };
-  stan::test::expect_common_unary_vectorized<stan::test::PromoteToComplex::Yes>(
-      f);
-  stan::test::expect_unary_vectorized<stan::test::PromoteToComplex::Yes>(
-      f, -15.2, -10, -0.5, 0.5, 1, 1.0, 1.3, 5, 10);
+  stan::test::expect_common_unary_vectorized<
+      stan::test::ScalarSupport::RealAndComplex>(f);
+  stan::test::expect_unary_vectorized<
+      stan::test::ScalarSupport::RealAndComplex>(f, -15.2, -10, -0.5, 0.5, 1,
+                                                 1.0, 1.3, 5, 10);
   stan::test::expect_complex_common(f);
 
   std::vector<double> com_args = stan::test::internal::common_nonzero_args();

test/unit/math/mix/fun/log1m_exp_test.cpp

@@ -3,7 +3,7 @@
 TEST(mathMixMatFun, log1m_exp) {
   auto f = [](const auto& x1) { return stan::math::log1m_exp(x1); };
   stan::test::expect_common_nonzero_unary_vectorized<
-      stan::test::PromoteToComplex::No>(f);
+      stan::test::ScalarSupport::Real>(f);
   stan::test::expect_unary_vectorized(f, -14, -12.6, -2, -1, -0.2, -0.5, 1.3,
                                       3);
 }