def test_value_with_one_input_returns_expected_value(self):
# Arrange
indicator = BollingerBands(20, 2.0)
# Act
indicator.update_raw(1.00020, 1.00000, 1.00010)
# Assert
assert indicator.upper == 1.00010
assert indicator.middle == 1.00010
assert indicator.lower == 1.00010
def test_value_with_three_inputs_returns_expected_value(self):
# Arrange
indicator = BollingerBands(20, 2.0)
# Act
indicator.update_raw(1.00020, 1.00000, 1.00015)
indicator.update_raw(1.00030, 1.00010, 1.00015)
indicator.update_raw(1.00040, 1.00020, 1.00021)
# Assert
assert indicator.upper == 1.0003155506390384
assert indicator.middle == 1.0001900000000001
assert indicator.lower == 1.0000644493609618
def test_initialized_with_required_inputs_returns_true(self):
# Arrange
indicator = BollingerBands(5, 2.0)
indicator.update_raw(1.00000, 1.00000, 1.00000)
indicator.update_raw(1.00000, 1.00000, 1.00000)
indicator.update_raw(1.00000, 1.00000, 1.00000)
indicator.update_raw(1.00000, 1.00000, 1.00000)
indicator.update_raw(1.00000, 1.00000, 1.00000)
# Act, Assert
assert indicator.initialized is True
def test_reset_successfully_returns_indicator_to_fresh_state(self):
# Arrange
indicator = BollingerBands(5, 2.0)
indicator.update_raw(1.00000, 1.00000, 1.00000)
indicator.update_raw(1.00000, 1.00000, 1.00000)
indicator.update_raw(1.00000, 1.00000, 1.00000)
indicator.update_raw(1.00000, 1.00000, 1.00000)
indicator.update_raw(1.00000, 1.00000, 1.00000)
# Act
indicator.reset()
# Assert
assert not indicator.initialized
assert indicator.upper == 0
assert indicator.middle == 0
assert indicator.lower == 0
def test_reset_successfully_returns_indicator_to_fresh_state(self):
# Arrange
indicator = BollingerBands(5, 2.0)
indicator.update_raw(1.00000, 1.00000, 1.00000)
indicator.update_raw(1.00000, 1.00000, 1.00000)
indicator.update_raw(1.00000, 1.00000, 1.00000)
indicator.update_raw(1.00000, 1.00000, 1.00000)
indicator.update_raw(1.00000, 1.00000, 1.00000)
# Act
indicator.reset()
# Assert
self.assertFalse(indicator.initialized)
self.assertEqual(0, indicator.upper)
self.assertEqual(0, indicator.middle)
self.assertEqual(0, indicator.lower)