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'])