henry000 commited on
Commit
1e4ec0a
Β·
1 Parent(s): 5c8e313

πŸ’š [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 == (480, 640) or origin_frame.size == (512, 640)
 
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)