11import string
22from io import StringIO
3- from random import Random
3+ from random import Random , choice
44
55import numpy as np
66import 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