Skip to content

Commit

Permalink
lint
Browse files Browse the repository at this point in the history
  • Loading branch information
wpbonelli committed Oct 4, 2023
1 parent 2a3b63c commit 833eafa
Show file tree
Hide file tree
Showing 2 changed files with 9 additions and 3 deletions.
4 changes: 3 additions & 1 deletion autotest/test_particledata.py
Original file line number Diff line number Diff line change
Expand Up @@ -309,7 +309,9 @@ def test_particledata_to_prp_disv_9():
def test_lrcparticledata_to_prp_divisions_defaults():
sd_data = CellDataType()
regions = [[0, 0, 1, 0, 1, 1]]
part_data = LRCParticleData(subdivisiondata=[sd_data], lrcregions=[regions])
part_data = LRCParticleData(
subdivisiondata=[sd_data], lrcregions=[regions]
)
grid = GridCases().structured_small()
rpts_prt = flatten(list(part_data.to_prp(grid)))
rpts_exp = [
Expand Down
8 changes: 6 additions & 2 deletions flopy/modpath/mp7particledata.py
Original file line number Diff line number Diff line change
Expand Up @@ -1109,7 +1109,9 @@ def to_coords(self, grid) -> Iterator[tuple]:
for i in range(mini, maxi + 1):
for j in range(minj, maxj + 1):
for sd in self.subdivisiondata:
for rpt in get_release_points(sd, grid, k, i, j):
for rpt in get_release_points(
sd, grid, k, i, j
):
yield (rpt[3], rpt[4], rpt[5])

def to_prp(self, grid) -> Iterator[tuple]:
Expand Down Expand Up @@ -1143,7 +1145,9 @@ def to_prp(self, grid) -> Iterator[tuple]:
for i in range(mini, maxi + 1):
for j in range(minj, maxj + 1):
for sd in self.subdivisiondata:
for irpt, rpt in enumerate(get_release_points(sd, grid, k, i, j)):
for irpt, rpt in enumerate(
get_release_points(sd, grid, k, i, j)
):
assert rpt[0] == k
assert rpt[1] == i
assert rpt[2] == j
Expand Down

0 comments on commit 833eafa

Please sign in to comment.