File tree Expand file tree Collapse file tree
dataprofiler/tests/labelers Expand file tree Collapse file tree Original file line number Diff line number Diff line change 11import logging
22import tempfile
33import unittest
4+ from pathlib import Path
45
56import numpy as np
67import pandas as pd
78import tensorflow as tf
89
9- from dataprofiler .labelers import labeler_utils
10+ from dataprofiler .labelers import labeler_utils , utils
1011
1112
1213class TestEvaluateAccuracy (unittest .TestCase ):
@@ -294,3 +295,17 @@ def test_hide_tf_logger_warnings(self):
294295 # make change and validate updated filter
295296 labeler_utils .hide_tf_logger_warnings ()
296297 self .assertEqual (1 + num_loggers , len (logger .filters ))
298+
299+
300+ class TestFindResourcesDir (unittest .TestCase ):
301+ def test_find_resources_dir (self ):
302+ resource_dir = utils .find_resources_dir ()
303+ self .assertTrue (resource_dir .is_dir ())
304+ self .assertEqual (Path ("resources" ).name , Path (str (resource_dir )).name )
305+
306+ labeler_dir = utils .find_resources_dir ("labelers" )
307+ self .assertTrue (labeler_dir .is_dir ())
308+ self .assertEqual (Path ("labelers" ).name , Path (str (labeler_dir )).name )
309+
310+ with self .assertRaisesRegex (FileNotFoundError , "Resource not found" ):
311+ utils .find_resources_dir ("does/not/exist" )
You can’t perform that action at this time.
0 commit comments