Skip to content

Commit

Permalink
fixing flake8
Browse files Browse the repository at this point in the history
  • Loading branch information
Casyfill committed May 10, 2023
1 parent d6c4a9e commit 02fbf61
Showing 1 changed file with 4 additions and 3 deletions.
7 changes: 4 additions & 3 deletions tests/test_v1.py
Original file line number Diff line number Diff line change
Expand Up @@ -23,9 +23,10 @@ def test_categorical_dtypes():
from dfschema.core.core import DfSchema
import json
from pathlib import Path
path = Path(__name__).parent / 'tests/test_schemas/v1/good/property_benchmarks.json'

path = Path(__name__).parent / "tests/test_schemas/v1/good/property_benchmarks.json"
schema = json.loads(path.read_text())

S = DfSchema.from_dict(schema)
catcol = [el for el in S.columns if el.name == 'BOROUGH_ID'][0]
catcol = [el for el in S.columns if el.name == "BOROUGH_ID"][0]
assert catcol.categorical.value_set == frozenset((100, 200, 300, 400, 500))

0 comments on commit 02fbf61

Please sign in to comment.