Skip to content

Commit

Permalink
More verbose tests
Browse files Browse the repository at this point in the history
  • Loading branch information
connortann committed Mar 7, 2024
1 parent 54fe015 commit 777b38f
Showing 1 changed file with 7 additions and 3 deletions.
10 changes: 7 additions & 3 deletions slicer/test_slicer.py
Original file line number Diff line number Diff line change
Expand Up @@ -246,6 +246,7 @@ def test_slicer_sparse():

candidates = [csc_array, csr_array, dok_array, lil_array]
for candidate in candidates:
print("testing:", type(candidate))
slicer = S(candidate)
actual = slicer[0, 0]
assert ctr_eq(actual.o, 1)
Expand Down Expand Up @@ -341,7 +342,8 @@ def test_operations_1d():
array = np.array(elements)
torch_array = torch.tensor(elements)
containers = [li, tup, array, torch_array, di, series]
for _, ctr in enumerate(containers):
for ctr in containers:
print("testing:", type(ctr))
slicer = AtomicSlicer(ctr)

assert ctr_eq(slicer[0], elements[0])
Expand Down Expand Up @@ -371,7 +373,8 @@ def test_operations_2d():
sparse_lil = lil_matrix(elements)

containers = [li, df, sparse_csc, sparse_csr, sparse_dok, sparse_lil]
for _, ctr in enumerate(containers):
for ctr in containers:
print("testing:", type(ctr))
slicer = AtomicSlicer(ctr)

assert ctr_eq(slicer[0], elements[0])
Expand Down Expand Up @@ -432,7 +435,8 @@ def test_operations_3d():
list_of_multi_arrays,
di_of_multi_arrays,
]
for _, ctr in enumerate(containers):
for ctr in containers:
print("testing:", type(ctr))
slicer = AtomicSlicer(ctr)

assert ctr_eq(slicer[0], elements[0])
Expand Down

0 comments on commit 777b38f

Please sign in to comment.