diff options
Diffstat (limited to 'test/sql/test_tablesample.py')
| -rw-r--r-- | test/sql/test_tablesample.py | 34 |
1 files changed, 19 insertions, 15 deletions
diff --git a/test/sql/test_tablesample.py b/test/sql/test_tablesample.py index 879e83182..712450d9f 100644 --- a/test/sql/test_tablesample.py +++ b/test/sql/test_tablesample.py @@ -16,10 +16,13 @@ class TableSampleTest(fixtures.TablesTest, AssertsCompiledSQL): @classmethod def define_tables(cls, metadata): - Table('people', metadata, - Column('people_id', Integer, primary_key=True), - Column('age', Integer), - Column('name', String(30))) + Table( + "people", + metadata, + Column("people_id", Integer, primary_key=True), + Column("age", Integer), + Column("name", String(30)), + ) def test_standalone(self): table1 = self.tables.people @@ -27,27 +30,28 @@ class TableSampleTest(fixtures.TablesTest, AssertsCompiledSQL): # no special alias handling even though clause is not in the # context of a FROM clause self.assert_compile( - tablesample(table1, 1, name='alias'), - 'people AS alias TABLESAMPLE system(:system_1)' + tablesample(table1, 1, name="alias"), + "people AS alias TABLESAMPLE system(:system_1)", ) self.assert_compile( - table1.tablesample(1, name='alias'), - 'people AS alias TABLESAMPLE system(:system_1)' + table1.tablesample(1, name="alias"), + "people AS alias TABLESAMPLE system(:system_1)", ) self.assert_compile( - tablesample(table1, func.bernoulli(1), name='alias', - seed=func.random()), - 'people AS alias TABLESAMPLE bernoulli(:bernoulli_1) ' - 'REPEATABLE (random())' + tablesample( + table1, func.bernoulli(1), name="alias", seed=func.random() + ), + "people AS alias TABLESAMPLE bernoulli(:bernoulli_1) " + "REPEATABLE (random())", ) def test_select_from(self): table1 = self.tables.people self.assert_compile( - select([table1.tablesample(text('1'), name='alias').c.people_id]), - 'SELECT alias.people_id FROM ' - 'people AS alias TABLESAMPLE system(1)' + select([table1.tablesample(text("1"), name="alias").c.people_id]), + "SELECT alias.people_id FROM " + "people AS alias TABLESAMPLE system(1)", ) |