Skip to content

Commit 8e29c28

Browse files
committed
fixed mypy errors
1 parent 5d5b473 commit 8e29c28

1 file changed

Lines changed: 6 additions & 6 deletions

File tree

tests/test_microdata.py

Lines changed: 6 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,6 @@
11
import string
22
from io import StringIO
3-
from random import Random
3+
from random import Random, choice
44

55
import numpy as np
66
import pandas as pd
@@ -42,7 +42,7 @@ def _tweak_safe_values_df(df: pd.DataFrame, values_to_tweak: list[int] = [29]) -
4242
# instance to a random value, thus ensuring that some 1dim values will
4343
# fail LCF, producing non-singularity leafs
4444
def ran_str10() -> str:
45-
return "".join(random.choice(string.ascii_letters) for i in range(10))
45+
return "".join(choice(string.ascii_letters) for i in range(10))
4646

4747
for column in df.columns:
4848
for value_to_tweak in values_to_tweak:
@@ -128,7 +128,7 @@ def test_generates_real_microdata() -> None:
128128
assert params.root_buffers.lower_low <= convertor.lower_buffer <= params.root_buffers.lower_high
129129
assert params.root_buffers.upper_low <= convertor.upper_buffer <= params.root_buffers.upper_high
130130

131-
microdata = generate_microdata(buckets, convertors, [1234.0, 1234.0], _rng)
131+
microdata = generate_microdata(buckets, cast(list[DataConvertor], convertors), [1234.0, 1234.0], _rng)
132132

133133
assert len(microdata) == 13
134134

@@ -158,7 +158,7 @@ def test_generates_bool_microdata() -> None:
158158
assert params.root_buffers.lower_low <= convertor.lower_buffer <= params.root_buffers.lower_high
159159
assert params.root_buffers.upper_low <= convertor.upper_buffer <= params.root_buffers.upper_high
160160

161-
microdata = generate_microdata(buckets, convertors, [1234.0, 1234.0], _rng)
161+
microdata = generate_microdata(buckets, cast(list[DataConvertor], convertors), [1234.0, 1234.0], _rng)
162162
for row in microdata:
163163
assert len(row) == 2
164164
for value in row:
@@ -179,7 +179,7 @@ def test_generates_int_microdata() -> None:
179179
assert params.root_buffers.lower_low <= convertor.lower_buffer <= params.root_buffers.lower_high
180180
assert params.root_buffers.upper_low <= convertor.upper_buffer <= params.root_buffers.upper_high
181181

182-
microdata = generate_microdata(buckets, convertors, [1234.0, 1234.0], _rng)
182+
microdata = generate_microdata(buckets, cast(list[DataConvertor], convertors), [1234.0, 1234.0], _rng)
183183
for row in microdata:
184184
assert len(row) == 2
185185
for value in row:
@@ -200,7 +200,7 @@ def test_generates_timestamp_microdata() -> None:
200200
assert params.root_buffers.lower_low <= convertor.lower_buffer <= params.root_buffers.lower_high
201201
assert params.root_buffers.upper_low <= convertor.upper_buffer <= params.root_buffers.upper_high
202202

203-
microdata = generate_microdata(buckets, convertors, [1234.0, 1234.0], _rng)
203+
microdata = generate_microdata(buckets, cast(list[DataConvertor], convertors), [1234.0, 1234.0], _rng)
204204
for row in microdata:
205205
assert len(row) == 2
206206
for value in row:

0 commit comments

Comments
 (0)