def test_cumulative(self):
class StockStub(ExprCtx):
def get_metrics(self, name: str, year: int, quarter: int):
return 1
stock = StockStub()
get = Get('cumulative')
v = get.eval(stock, 2019, 1)
self.assertEqual(1, v)
print(get, '\r\n')
def test_increment_q4(self):
class StockStub(ExprCtx):
def get_metrics(self, name: str, year: int, quarter: int):
if year == 2019 and quarter == 4:
return 3
elif year == 2019 and quarter == 3:
return 1
stock = StockStub()
get = Get('increment', increment=True)
v = get.eval(stock, 2019, 4)
self.assertEqual(2, v)
print(get, '\r\n')
