inst/tinytest/test_timeOHLCV.R

In bobjansen/FancyBars: Calculate Various Types of OHLC Bars

library(data.table)

# One trade ----
jan1_2020 <- as.POSIXct('2020-01-01', tz = 'UTC')
jan2_2020 <- as.POSIXct('2020-01-02', tz = 'UTC')
jan3_2020 <- as.POSIXct('2020-01-03', tz = 'UTC')
jan4_2020 <- as.POSIXct('2020-01-04', tz = 'UTC')
test_trades <- data.table(
  Timestamp = jan1_2020,
  Symbol = 'AAA',
  Price = 100,
  Size = 10L
)
bar <- timeOHLCV(test_trades)
expect_equal(names(bar), c(
  'Timestamp', 'Open', 'High', 'Low', 'Close', 'Volume', 'VWAP', 'TickCount')
)
expect_equal(bar[, Open], 100)
expect_equal(bar[, Low], 100)
expect_equal(bar[, High], 100)
expect_equal(bar[, Close], 100)
expect_equal(bar[, VWAP], 100)
expect_equal(bar[, Volume], 10L)
expect_equal(bar[, TickCount], 1L)

# Two trade ----
test_trades <- data.table(
  Timestamp = c(jan1_2020, jan2_2020),
  Symbol = 'AAA',
  Price = c(100, 110),
  Size = c(10L, 20L)
)
bar <- timeOHLCV(test_trades, align_by = 'days', align_period = 2L)
expect_equal(names(bar), c(
  'Timestamp', 'Open', 'High', 'Low', 'Close', 'Volume', 'VWAP', 'TickCount')
)
expect_equal(bar[, Open], 100)
expect_equal(bar[, Low], 100)
expect_equal(bar[, High], 110)
expect_equal(bar[, Close], 110)
expect_equal(bar[, VWAP], 320 / 3)
expect_equal(bar[, Volume], 30L)
expect_equal(bar[, TickCount], 2L)

# Four trade ----
test_trades <- data.table(
  Timestamp = c(jan1_2020, jan2_2020, jan3_2020, jan4_2020),
  Symbol = 'AAA',
  Price = c(120, 100, 110, 105),
  Size = c(10L, 20L, 30L, 40L)
)
bar <- timeOHLCV(test_trades, align_by = 'days', align_period = 2L)
expect_equal(names(bar), c(
  'Timestamp', 'Open', 'High', 'Low', 'Close', 'Volume', 'VWAP', 'TickCount')
)
expect_equal(bar[1L, Open], 120)
expect_equal(bar[1L, Low], 100)
expect_equal(bar[1L, High], 120)
expect_equal(bar[1L, Close], 100)
expect_equal(bar[1L, VWAP], 320 / 3)
expect_equal(bar[1L, Volume], 30L)
expect_equal(bar[1L, TickCount], 2L)

expect_equal(bar[2L, Open], 110)
expect_equal(bar[2L, Low], 105)
expect_equal(bar[2L, High], 110)
expect_equal(bar[2L, Close], 105)
expect_equal(bar[2L, VWAP], 750 / 7)
expect_equal(bar[2L, Volume], 70L)
expect_equal(bar[2L, TickCount], 2L)

# Incomplete bar ----
jan1_2020_six <- as.POSIXct('2020-01-01 06:00:00', tz = 'UTC')
jan1_2020 <- as.POSIXct('2020-01-01 12:00:00', tz = 'UTC')
jan2_2020 <- as.POSIXct('2020-01-02 12:00:00', tz = 'UTC')
jan3_2020 <- as.POSIXct('2020-01-03 12:00:00', tz = 'UTC')
jan4_2020 <- as.POSIXct('2020-01-04 12:00:00', tz = 'UTC')


test_trades <- data.table(
  Timestamp = jan1_2020,
  Symbol = 'AAA',
  Price = 100,
  Size = 10L
)
bar <- timeOHLCV(
  test_trades, align_by = 'days', align_period = 1L, name = 'AAA'
)
test_trades <- data.table(
  Timestamp = c(jan1_2020, jan2_2020, jan3_2020, jan4_2020),
  Symbol = 'AAA',
  Price = c(120, 100, 110, 105),
  Size = c(10L, 20L, 30L, 40L)
)
bar <- timeOHLCV(
  test_trades, align_by = 'days', align_period = 1L, name = 'AAA',
  prev_bar = bar
)