mlir/unittests/Analysis/Presburger/FractionTest.cpp

   1 #include "mlir/Analysis/Presburger/Fraction.h"
   2 #include "./Utils.h"
   3 #include <gmock/gmock.h>
   4 #include <gtest/gtest.h>
   5
   6 using namespace mlir;
   7 using namespace presburger;
   8
   9 TEST(FractionTest, getAsInteger) {
  10   Fraction f(3, 1);
  11   EXPECT_EQ(f.getAsInteger(), DynamicAPInt(3));
  12 }
  13
  14 TEST(FractionTest, nearIntegers) {
  15   Fraction f(52, 14);
  16
  17   EXPECT_EQ(floor(f), 3);
  18   EXPECT_EQ(ceil(f), 4);
  19 }
  20
  21 TEST(FractionTest, reduce) {
  22   Fraction f(20, 35), g(-56, 63);
  23   EXPECT_EQ(f, Fraction(4, 7));
  24   EXPECT_EQ(g, Fraction(-8, 9));
  25 }
  26
  27 TEST(FractionTest, arithmetic) {
  28   Fraction f(3, 4), g(-2, 3);
  29
  30   EXPECT_EQ(f / g, Fraction(-9, 8));
  31   EXPECT_EQ(f * g, Fraction(-1, 2));
  32   EXPECT_EQ(f + g, Fraction(1, 12));
  33   EXPECT_EQ(f - g, Fraction(17, 12));
  34
  35   f /= g;
  36   EXPECT_EQ(f, Fraction(-9, 8));
  37   f *= g;
  38   EXPECT_EQ(f, Fraction(3, 4));
  39   f += g;
  40   EXPECT_EQ(f, Fraction(Fraction(1, 12)));
  41   f -= g;
  42   EXPECT_EQ(f, Fraction(3, 4));
  43 }
  44
  45 TEST(FractionTest, relational) {
  46   Fraction f(2, 5), g(3, 7);
  47   EXPECT_TRUE(f < g);
  48   EXPECT_FALSE(g < f);
  49
  50   EXPECT_EQ(f, Fraction(4, 10));
  51 }