datasets

1
import unittest
2
import warnings
3

4
from datasets.utils import experimental
5

6

7
@experimental
8
def dummy_function():
9
    return "success"
10

11

12
class TestExperimentalFlag(unittest.TestCase):
13
    def test_experimental_warning(self):
14
        with warnings.catch_warnings(record=True) as w:
15
            warnings.simplefilter("always")
16
            self.assertEqual(dummy_function(), "success")
17
        self.assertEqual(len(w), 1)
18