#include <quiz.h>
#include <poincare.h>
#include <assert.h>

#if POINCARE_TESTS_PRINT_EXPRESSIONS
#include "../src/expression_debug.h"
#include <iostream>
using namespace std;
#endif

void assert_simplifies_to(const char * input_string, const char * expected_string) {
#if POINCARE_TESTS_PRINT_EXPRESSIONS
  cout << "---- Simplification Run ----"  << endl;
  cout << input_string << " -> " << expected_string << endl;
#endif
  //Expression* tab[3] = {new Integer(1), new Integer(2), new Integer(3)};
  //Expression* input = new Addition(tab, 3, false);
  Expression * input = Expression::parse(input_string);
  assert(input != nullptr);
#if POINCARE_TESTS_PRINT_EXPRESSIONS
  cout << "Input = " << endl;
  print_expression(input);
#endif

  Expression * simplified = input->simplify();
  assert(simplified != nullptr);
#if POINCARE_TESTS_PRINT_EXPRESSIONS
  cout << "Simplified = " << endl;
  print_expression(simplified);
#endif

  Expression * expected = Expression::parse(expected_string);
  assert(expected != nullptr);
#if POINCARE_TESTS_PRINT_EXPRESSIONS
  cout << "Expected = " << endl;
  print_expression(expected);
#endif

  assert(simplified->isIdenticalTo(expected));
  delete expected;
  if (simplified != input) {
    delete simplified;
  }
  delete input;
}

QUIZ_CASE(poincare_simplify_addition_integer) {
  assert_simplifies_to("1", "1");
  assert_simplifies_to("1+2", "3");
  assert_simplifies_to("1+a", "1+a");
  assert_simplifies_to("1+2+3+4+5+6+7", "28");
  assert_simplifies_to("1+2+3+4+5+a+6+7", "28+a");
}