def test_cmf_invalid_data(self): self.close_data.append(1) period = 6 with self.assertRaises(Exception) as cm: chaikin_money_flow.chaikin_money_flow(self.close_data, self.high_data, self.low_data, self.volume, period) expected = ("Error: mismatched data lengths, check to ensure that all input data is the same length and valid") self.assertEqual(str(cm.exception), expected)
def cmf(dataframe, period=14) -> ndarray: from pyti.chaikin_money_flow import chaikin_money_flow return chaikin_money_flow(dataframe['close'], dataframe['high'], dataframe['low'], dataframe['volume'], period)
def test_cmf_invalid_period(self): period = 128 with self.assertRaises(Exception) as cm: chaikin_money_flow.chaikin_money_flow(self.close_data, self.high_data, self.low_data, self.volume, period) expected = "Error: data_len < period" self.assertEqual(str(cm.exception), expected)
def test_cmf_period_10(self): period = 10 cmf = chaikin_money_flow.chaikin_money_flow(self.close_data, self.high_data, self.low_data, self.volume, period) np.testing.assert_array_equal(cmf, self.cmf_period_10_expected)