require 'spec_helper'

describe Qlang do
  describe 'Function' do
    context 'into R' do
      it do
        expect(
          Q.to_r.compile('f(x, y) = x + y')
        ).to eq(
          "f <- function(x, y) x + y"
        )

        expect(
          Q.to_r.compile('g(x) = x ^ 2')
        ).to eq(
          "g <- function(x) x ^ 2"
        )

        expect(
          Q.to_r.compile('g(x) = x ^ (2 + 2)')
        ).to eq(
          "g <- function(x) x ^ (2 + 2)"
        )

        expect(
          Q.to_r.compile('h(a, b, c) = a ^ 2 + b ^ 2 + c ^ 2')
        ).to eq(
          "h <- function(a, b, c) a ^ 2 + b ^ 2 + c ^ 2"
        )
      end
    end
  end
end