def testShortTimeSpanAverage(self):
# build up a small time series
local_series = SFATimeSeries()
local_series.append(1)
time.sleep(0.1)
local_series.append(2)
time.sleep(0.1)
local_series.append(3)
result = local_series.average(0.0001)
def testFlush(self):
# Note: make sure testAppend runs first because we need append() for this test
SERIES_SIZE = 50
local_series = SFATimeSeries()
for i in range(SERIES_SIZE):
local_series.append(1)
self.assertEqual(local_series.size(), SERIES_SIZE,
"Expected the series to be full. It wasn't.")
local_series.flush()
self.assertEqual(
local_series.size(), 0,
"Expected the series to be empty. It contained %d items." %
local_series.size())
def testAppend(self):
SERIES_SIZE = 500 # Number of values to append to the series
START_VALUE = 10
INCREMENT = 1
local_series = SFATimeSeries(SERIES_SIZE)
value = START_VALUE
while local_series.size() < SERIES_SIZE:
orig_size = local_series.size()
local_series.append(value)
value += INCREMENT
self.assertEqual(
orig_size + 1, local_series.size(),
"Append to series failed. Series size = %d" %
local_series.size())
# series is now full. Verify first & last elements
self.assertEqual(
local_series.size(), SERIES_SIZE,
"Series was not full when it should have been. Actual size: %d Expected size: %d"
% (local_series.size(), SERIES_SIZE))
self.assertEqual(
local_series.get(0)[0], START_VALUE,
"Unexpected first value in the series. Did we remove it during an append?"
)
self.assertEqual(
local_series.get(SERIES_SIZE - 1)[0], value - INCREMENT,
"Unexpected first value in the series. Did we remove it during an append?"
)
# Add one more value and make sure the size stays at its max
local_series.append(value)
value += INCREMENT
self.assertEqual(
local_series.size(), SERIES_SIZE,
"Unexpected series size. Actual size: %d Expected size: %d" %
(local_series.size(), SERIES_SIZE))
self.assertEqual(
local_series.get(0)[0], START_VALUE + INCREMENT,
"Unexpected first value in the series. Did we remove it during an append?"
)
self.assertEqual(
local_series.get(SERIES_SIZE - 1)[0], value - INCREMENT,
"Unexpected first value in the series. Did we remove it during an append?"
)
def generate_time_series(self):
# Build up a time series that we can mess with...
print
print "Generating time series data. This will take approximately %d seconds." % SERIES_LENGTH
self._series = SFATimeSeries()
approx_inter_item_time = SERIES_LENGTH / SERIES_SIZE
start_time = time.time()
append_val = 1
self._series.append(append_val)
while time.time() < (start_time + SERIES_LENGTH):
last_append_time = self._series.get(self._series.size() - 1)[1]
while time.time() < (last_append_time + approx_inter_item_time):
time.sleep(approx_inter_item_time / 100)
append_val += SERIES_RATE * (time.time() - last_append_time)
self._series.append(append_val)
print "Time series generated."
def testSingleValueAverage(self):
local_series = SFATimeSeries()
self.assertRaises(EmptyTimeSeriesException, local_series.average, 1)
VALUE = 25
local_series.append(VALUE)
self.assertRaises(EmptyTimeSeriesException, local_series.average, 1)
