|
1 | | -import math |
2 | 1 | import numpy as np |
3 | 2 | import subprocess |
4 | 3 | import os |
5 | 4 | import pandas as pd |
6 | 5 | import pyarrow as pa |
7 | | -import re |
8 | 6 | import glob |
9 | 7 | import shutil |
10 | 8 | import platform |
@@ -41,7 +39,7 @@ def check_if_compatible(uri): |
41 | 39 | try: |
42 | 40 | with tiledb.open(uri): |
43 | 41 | return True |
44 | | - except tiledb.cc.TileDBError as e: |
| 42 | + except tiledb.libtiledb.TileDBError as e: |
45 | 43 | if "incompatible format version" in str(e).lower(): |
46 | 44 | raise pytest.skip.Exception( |
47 | 45 | "Test skipped due to incompatible format version" |
@@ -127,7 +125,9 @@ def test_retrieve_samples(test_ds): |
127 | 125 |
|
128 | 126 | def test_read_unsupported_regions_type(test_ds): |
129 | 127 | unsupported_region = 3.14 |
130 | | - unsupported_type_error = f'"regions" parameter cannot have type: {type(unsupported_region)}' |
| 128 | + unsupported_type_error = ( |
| 129 | + f'"regions" parameter cannot have type: {type(unsupported_region)}' |
| 130 | + ) |
131 | 131 | wrong_dimension_region = np.array([["1:12700-13400"], ["1:12700-13400"]]) |
132 | 132 | ndarray_wrong_dimension_error = f'"regions" parameter of type {type(wrong_dimension_region)} must be 1-dimensional' |
133 | 133 | with pytest.raises(Exception, match=unsupported_type_error): |
@@ -277,13 +277,15 @@ def test_basic_reads(test_ds): |
277 | 277 |
|
278 | 278 | # Regions as numpy.ndarray |
279 | 279 | df = test_ds.read( |
280 | | - attrs=["sample_name", "pos_start", "pos_end"], regions=np.array(["1:12700-13400"]) |
| 280 | + attrs=["sample_name", "pos_start", "pos_end"], |
| 281 | + regions=np.array(["1:12700-13400"]), |
281 | 282 | ) |
282 | 283 | _check_dfs( |
283 | 284 | expected_df, df.sort_values(ignore_index=True, by=["sample_name", "pos_start"]) |
284 | 285 | ) |
285 | 286 | df = test_ds.read_arrow( |
286 | | - attrs=["sample_name", "pos_start", "pos_end"], regions=np.array(["1:12700-13400"]) |
| 287 | + attrs=["sample_name", "pos_start", "pos_end"], |
| 288 | + regions=np.array(["1:12700-13400"]), |
287 | 289 | ).to_pandas() |
288 | 290 | _check_dfs( |
289 | 291 | expected_df, df.sort_values(ignore_index=True, by=["sample_name", "pos_start"]) |
@@ -438,12 +440,12 @@ def test_incomplete_read_generator(): |
438 | 440 | cfg = tiledbvcf.ReadConfig(memory_budget_mb=0) |
439 | 441 | test_ds = tiledbvcf.Dataset(uri, mode="r", cfg=cfg) |
440 | 442 | expected_df = pd.DataFrame.from_dict( |
441 | | - { |
442 | | - "pos_end": np.array( |
443 | | - [12771, 12771, 13374, 13389, 13395, 13413], dtype=np.int32 |
444 | | - ) |
445 | | - } |
446 | | - ) |
| 443 | + { |
| 444 | + "pos_end": np.array( |
| 445 | + [12771, 12771, 13374, 13389, 13395, 13413], dtype=np.int32 |
| 446 | + ) |
| 447 | + } |
| 448 | + ) |
447 | 449 |
|
448 | 450 | # NOTE: Running multiple test shows that the iterator can be reused |
449 | 451 |
|
|
0 commit comments