nvim-treesitter/tests/indent/r_spec.lua

local Runner = require('tests.indent.common').Runner
--local XFAIL = require("tests.indent.common").XFAIL

local run = Runner:new(it, 'tests/indent/r', {
  tabstop = 2,
  shiftwidth = 2,
  softtabstop = 0,
  expandtab = true,
})

describe('indent R:', function()
  describe('whole file:', function()
    run:whole_file('.', {})
  end)

  describe('new line:', function()
    run:new_line('comment.R', { on_line = 1, text = '# new comment', indent = 0 })

    run:new_line('cond.R', { on_line = 1, text = 'x <- x + 1', indent = 0 })
    run:new_line('cond.R', { on_line = 4, text = 'x <- x + 1', indent = 2 })
    run:new_line('cond.R', { on_line = 5, text = 'x <- x + 1', indent = 2 })
    run:new_line('cond.R', { on_line = 8, text = 'x <- x + 1', indent = 4 })

    run:new_line('func.R', { on_line = 1, text = 'x <- x + 1', indent = 2 })
    run:new_line('func.R', { on_line = 2, text = 'a <- a + 1', indent = 4 })
    run:new_line('func.R', { on_line = 6, text = '0,', indent = 6 })
    run:new_line('func.R', { on_line = 6, text = '0,', indent = 6 })
    run:new_line('func.R', { on_line = 16, text = 'x <- x + 1', indent = 4 })

    run:new_line('loop.R', { on_line = 1, text = 'x <- x + 1', indent = 0 })
    run:new_line('loop.R', { on_line = 3, text = 'x <- x + 1', indent = 2 })
    run:new_line('loop.R', { on_line = 8, text = 'x <- x + 1', indent = 2 })
    run:new_line('loop.R', { on_line = 14, text = "print('lol')", indent = 4 })

    run:new_line('pipe.R', { on_line = 1, text = 'head(n = 10L) |>', indent = 2 })
    run:new_line('pipe.R', { on_line = 9, text = 'head()', indent = 2 })

    run:new_line('incomplete_pipe.R', { on_line = 2, text = 'head %>%', indent = 2 })

    run:new_line('aligned_indent.R', { on_line = 1, text = 'z,', indent = 17 })
  end)
end)