コード例 #1
0
ファイル: test_yql.py プロジェクト: stjordanis/yfi
 def setUp(self):
     self.yql = Yql()
コード例 #2
0
ファイル: test_yql.py プロジェクト: nickelkr/yfi
 def setUp(self):
     self.yql = Yql()
コード例 #3
0
ファイル: test_yql.py プロジェクト: stjordanis/yfi
class TestYQL(unittest.TestCase):
    def setUp(self):
        self.yql = Yql()

    def test_single_select(self):
        self.yql.select('*')
        self.assertIn('select * from yahoo.finance.quotes', self.yql.parts())

    def test_set_table(self):
        self.yql.set_table('yahoo.finance.historicaldata')
        self.assertEqual('yahoo.finance.historicaldata', self.yql.get_table())

    def test_set_endpoint(self):
        self.yql.set_endpoint('/just/a/test')
        self.assertEqual('/just/a/test', self.yql.get_endpoint())

    def test_set_store(self):
        self.yql.set_store('adiffstore.com/all')
        self.assertEqual('store://adiffstore.com/all', self.yql.get_store())

    def test_xml(self):
        self.yql.set_format('xml')
        self.assertEqual('xml', self.yql.get_format())

    def test_multi_select(self):
        self.yql.select('Ask', 'Bid')
        self.assertIn('select Ask, Bid from yahoo.finance.quotes',
                      self.yql.parts())

    def test_select_all(self):
        self.yql.select()
        self.assertIn('select * from yahoo.finance.quotes', self.yql.parts())

    def test_single_where(self):
        self.yql.where('symbol')
        self.assertIn('where symbol', self.yql.parts())

    def test_single__in(self):
        self.yql._in('TSLA')
        self.assertIn('in ("TSLA")', self.yql.parts())

    def test_multi_in(self):
        self.yql._in('TSLA', 'GOOG')
        self.assertIn('in ("TSLA", "GOOG")', self.yql.parts())

    def test_and(self):
        self.yql._and()
        self.assertIn('and', self.yql.parts())

    def test_chaining(self):
        self.yql.select('*').where('symbol')._in('TSLA')
        self.assertIn('select * from yahoo.finance.quotes', self.yql.parts())
        self.assertIn('where symbol', self.yql.parts())
        self.assertIn('in ("TSLA")', self.yql.parts())

    def test_sequence(self):
        self.yql.select('*')
        self.yql.where('symbol')
        self.yql._in('TSLA')
        self.assertIn('select * from yahoo.finance.quotes', self.yql.parts())
        self.assertIn('where symbol', self.yql.parts())
        self.assertIn('in ("TSLA")', self.yql.parts())

    def test_symbol(self):
        self.yql.symbol('TSLA')
        self.assertIn('select * from yahoo.finance.quotes', self.yql.parts())
        self.assertIn('where symbol', self.yql.parts())
        self.assertIn('in ("TSLA")', self.yql.parts())

    def test_equal(self):
        self.yql.eq('92.60')
        self.assertIn('= 92.60', self.yql.parts())
コード例 #4
0
ファイル: test_yql.py プロジェクト: nickelkr/yfi
class TestYQL(unittest.TestCase):

    def setUp(self):
        self.yql = Yql()

    def test_single_select(self):
        self.yql.select('*')
        self.assertIn('select * from yahoo.finance.quotes', self.yql.parts())

    def test_set_table(self):
        self.yql.set_table('yahoo.finance.historicaldata')
        self.assertEqual('yahoo.finance.historicaldata', self.yql.get_table())

    def test_set_endpoint(self):
        self.yql.set_endpoint('/just/a/test')
        self.assertEqual('/just/a/test', self.yql.get_endpoint())

    def test_set_store(self):
        self.yql.set_store('adiffstore.com/all')
        self.assertEqual('store://adiffstore.com/all', self.yql.get_store())

    def test_xml(self):
        self.yql.set_format('xml')
        self.assertEqual('xml', self.yql.get_format())

    def test_multi_select(self):
        self.yql.select('Ask', 'Bid')
        self.assertIn('select Ask, Bid from yahoo.finance.quotes', self.yql.parts())

    def test_select_all(self):
        self.yql.select()
        self.assertIn('select * from yahoo.finance.quotes', self.yql.parts())

    def test_single_where(self):
        self.yql.where('symbol')
        self.assertIn('where symbol', self.yql.parts())

    def test_single__in(self):
        self.yql._in('TSLA')
        self.assertIn('in ("TSLA")', self.yql.parts())

    def test_multi_in(self):
        self.yql._in('TSLA', 'GOOG')
        self.assertIn('in ("TSLA", "GOOG")', self.yql.parts())

    def test_and(self):
        self.yql._and()
        self.assertIn('and', self.yql.parts())

    def test_chaining(self):
        self.yql.select('*').where('symbol')._in('TSLA')
        self.assertIn('select * from yahoo.finance.quotes', self.yql.parts())
        self.assertIn('where symbol', self.yql.parts())
        self.assertIn('in ("TSLA")', self.yql.parts())

    def test_sequence(self):
        self.yql.select('*')
        self.yql.where('symbol')
        self.yql._in('TSLA')
        self.assertIn('select * from yahoo.finance.quotes', self.yql.parts())
        self.assertIn('where symbol', self.yql.parts())
        self.assertIn('in ("TSLA")', self.yql.parts())

    def test_symbol(self):
        self.yql.symbol('TSLA')
        self.assertIn('select * from yahoo.finance.quotes', self.yql.parts())
        self.assertIn('where symbol', self.yql.parts())
        self.assertIn('in ("TSLA")', self.yql.parts())

    def test_equal(self):
        self.yql.eq('92.60')
        self.assertIn('= 92.60', self.yql.parts())