#include <stan/math/prim/mat.hpp>
#include <gtest/gtest.h>

template <typename T>
void test_rank() {
  using stan::math::rank;

  T c(1);
  c[0] = 1.7;
  EXPECT_EQ(0U, rank(c,1));
  EXPECT_THROW(rank(c,0), std::out_of_range);
  EXPECT_THROW(rank(c,2), std::out_of_range);

  T e(2);
  e[0] = 5.9; e[1] = -1.2;
  EXPECT_EQ(1U, rank(e,1));
  EXPECT_EQ(0U, rank(e,2));
  EXPECT_THROW(rank(e,0), std::out_of_range);
  EXPECT_THROW(rank(e,3), std::out_of_range);

  T g(3);
  g[0] = 5.9; g[1] = -1.2; g[2] = 192.13456;
  EXPECT_EQ(1U, rank(g,1));
  EXPECT_EQ(0U, rank(g,2));
  EXPECT_EQ(2U, rank(g,3));
  EXPECT_THROW(rank(g,0), std::out_of_range);
  EXPECT_THROW(rank(g,4), std::out_of_range);
  
  T z;
  EXPECT_THROW(rank(z,0), std::out_of_range);
  EXPECT_THROW(rank(z,1), std::out_of_range);
  EXPECT_THROW(rank(z,2), std::out_of_range); 
}


template <typename T>
void test_rank_int() {
  using stan::math::rank;

  T c(1);
  c[0] = 1;
  EXPECT_EQ(0U, rank(c,1));
  EXPECT_THROW(rank(c,0), std::out_of_range);
  EXPECT_THROW(rank(c,2), std::out_of_range);

  T e(2);
  e[0] = 5; e[1] = -1;
  EXPECT_EQ(1U, rank(e,1));
  EXPECT_EQ(0U, rank(e,2));
  EXPECT_THROW(rank(e,0), std::out_of_range);
  EXPECT_THROW(rank(e,3), std::out_of_range);

  T g(3);
  g[0] = 5; g[1] = -1; g[2] = 192;
  EXPECT_EQ(1U, rank(g,1));
  EXPECT_EQ(0U, rank(g,2));
  EXPECT_EQ(2U, rank(g,3));
  EXPECT_THROW(rank(g,0), std::out_of_range);
  EXPECT_THROW(rank(g,4), std::out_of_range);
  
  T z;
  EXPECT_THROW(rank(z,0), std::out_of_range);
  EXPECT_THROW(rank(z,1), std::out_of_range);
  EXPECT_THROW(rank(z,2), std::out_of_range); 
}

TEST(MathMatrix,rank) {
  using stan::math::rank;
  
  test_rank< std::vector<double> >();
  test_rank< Eigen::Matrix<double,Eigen::Dynamic,1> >();
  test_rank< Eigen::Matrix<double,1,Eigen::Dynamic> >();

  test_rank_int< std::vector<int> >();
  test_rank_int< std::vector<double> >();
  test_rank_int< Eigen::Matrix<double,Eigen::Dynamic,1> >();
  test_rank_int< Eigen::Matrix<double,1,Eigen::Dynamic> >();
}
