π [Fix] test bugs, pass streamloader orgimg shape
Browse files
tests/test_tools/test_data_loader.py
CHANGED
@@ -66,4 +66,4 @@ def test_directory_stream_data_loader_frame(directory_stream_data_loader: Stream
|
|
66 |
frame, rev_tensor, origin_frame = next(iter(directory_stream_data_loader))
|
67 |
assert frame.shape == (1, 3, 640, 640)
|
68 |
assert rev_tensor.shape == (1, 5)
|
69 |
-
assert origin_frame.size
|
|
|
66 |
frame, rev_tensor, origin_frame = next(iter(directory_stream_data_loader))
|
67 |
assert frame.shape == (1, 3, 640, 640)
|
68 |
assert rev_tensor.shape == (1, 5)
|
69 |
+
assert origin_frame.size != (640, 640)
|