From 32ceec8f74d5ac2a99a99d9c8a3110c5f47ebaad Mon Sep 17 00:00:00 2001
From: "xingjun.wang" <xingjun.wxj@alibaba-inc.com>
Date: Sun, 21 Jul 2024 19:13:37 +0800
Subject: [PATCH] add assert for youku data UT

---
 tests/msdatasets/test_general_datasets.py | 6 +++++-
 1 file changed, 5 insertions(+), 1 deletion(-)

diff --git a/tests/msdatasets/test_general_datasets.py b/tests/msdatasets/test_general_datasets.py
index a3a21609..0571fdcd 100644
--- a/tests/msdatasets/test_general_datasets.py
+++ b/tests/msdatasets/test_general_datasets.py
@@ -1,4 +1,5 @@
 # Copyright (c) Alibaba, Inc. and its affiliates.
+import os
 import unittest
 
 from modelscope import MsDataset
@@ -101,7 +102,10 @@ class GeneralMsDatasetTest(unittest.TestCase):
             use_streaming=True)
 
         logger.info(next(iter(ds)))
-        assert next(iter(ds))
+        data_sample = next(iter(ds))
+
+        assert data_sample['video_id']
+        assert os.path.exists(data_sample['video_id:FILE'])
 
 
 if __name__ == '__main__':