summaryrefslogtreecommitdiffstats
path: root/tests/test_prioritization.py
diff options
context:
space:
mode:
authorIrina Truong <i.chernyavska@gmail.com>2019-05-25 13:08:56 -0700
committerGitHub <noreply@github.com>2019-05-25 13:08:56 -0700
commit8cb7009bcd0f0062942932c853706a36178f566c (patch)
treecb5bb42674ccce90b173e7a2f09bf72157ae4a86 /tests/test_prioritization.py
parenta5e607b6fc889afd3f8960ca3903ae16b641c304 (diff)
black all the things. (#1049)
* added black to develop guide * no need for pep8radius. * changelog. * Add pre-commit checkbox. * Add pre-commit to dev reqs. * Add pyproject.toml for black. * Pre-commit config. * Add black to travis and dev reqs. * Install and run black in travis. * Remove black from dev reqs. * Lower black target version. * Re-format with black.
Diffstat (limited to 'tests/test_prioritization.py')
-rw-r--r--tests/test_prioritization.py10
1 files changed, 5 insertions, 5 deletions
diff --git a/tests/test_prioritization.py b/tests/test_prioritization.py
index 3046456e..f5b67006 100644
--- a/tests/test_prioritization.py
+++ b/tests/test_prioritization.py
@@ -3,18 +3,18 @@ from pgcli.packages.prioritization import PrevalenceCounter
def test_prevalence_counter():
counter = PrevalenceCounter()
- sql = '''SELECT * FROM foo WHERE bar GROUP BY baz;
+ sql = """SELECT * FROM foo WHERE bar GROUP BY baz;
select * from foo;
SELECT * FROM foo WHERE bar GROUP
- BY baz'''
+ BY baz"""
counter.update(sql)
- keywords = ['SELECT', 'FROM', 'GROUP BY']
+ keywords = ["SELECT", "FROM", "GROUP BY"]
expected = [3, 3, 2]
kw_counts = [counter.keyword_count(x) for x in keywords]
assert kw_counts == expected
- assert counter.keyword_count('NOSUCHKEYWORD') == 0
+ assert counter.keyword_count("NOSUCHKEYWORD") == 0
- names = ['foo', 'bar', 'baz']
+ names = ["foo", "bar", "baz"]
name_counts = [counter.name_count(x) for x in names]
assert name_counts == [3, 2, 2]