class TestYchartsProvider():
"""Tests for the YchartsProvider class."""
@test(groups=['setup'])
def setup(self):
"""Setup a YchartsPovider object for testing"""
self.provider = YchartsProvider('GOOG', StringIO(GOOG_DATA), ['free_cash_flow'])
@test(depends_on_groups=['setup'])
def test_get_all_before(self):
"""Test the get_all_before method."""
results = self.provider.get_all_before('free_cash_flow', date(2010, 1, 1))
assert_equal(len(results), 26)
assert_equal(results[0], 116621000.0)
assert_equal(results[-1], 2539000000.0)
@test(depends_on_groups=['setup'])
def test_get_most_recent(self):
"""Test the get_most_recent method."""
result = self.provider.get_most_recent('free_cash_flow', date(2010, 1, 1))
assert_equal(result, 2539000000.0)
@test(depends_on_groups=['setup'])
def test_get_next_date(self):
"""Test the get_next_date method."""
result = self.provider.get_next_date(date(2011, 10, 1))
assert_equal(result, date(2011, 10, 27))
@test(depends_on_groups=['setup'])
def get_first_date(self):
"""Test the get_first_date method."""
result = self.provider.get_first_date()
assert_equal(result, date(2004,8,17))
def setup(self):
"""Setup a YchartsPovider object for testing"""
self.provider = YchartsProvider('GOOG', StringIO(GOOG_DATA), ['free_cash_flow'])
