Find nested Faker provider submodules

rdt/transformers/pii/anonymizer.py

@@ -5,6 +5,7 @@
 import inspect
 import warnings
 from copy import deepcopy
+from operator import attrgetter
 
 import faker
 import numpy as np
@@ -51,7 +52,7 @@ def check_provider_function(provider_name, function_name):
             ``AttributeError`` if the provider or the function is not found.
         """
         try:
-            module = getattr(faker.providers, provider_name)
+            module = attrgetter(provider_name)(faker.providers)
             if provider_name.lower() == 'baseprovider':
                 getattr(module, function_name)
 

tests/unit/transformers/pii/test_anonymizer.py

@@ -19,42 +19,47 @@ class TestAnonymizedFaker:
 
     @patch('rdt.transformers.pii.anonymizer.faker')
     @patch('rdt.transformers.pii.anonymizer.getattr')
-    def test_check_provider_function_baseprovider(self, mock_getattr, mock_faker):
+    @patch('rdt.transformers.pii.anonymizer.attrgetter')
+    def test_check_provider_function_baseprovider(self, mock_attrgetter, mock_getattr, mock_faker):
         """Test that ``getattr`` is being called with ``BaseProvider`` and ``function_name``.
 
         Mock:
             - Mock the ``getattr`` from Python to ensure that is being called with the input.
             - Mock faker and ensure that ``getattr`` is being called with ``faker.providers``.
         """
         # Setup
-        mock_getattr.side_effect = ['module', 'provider', None]
+        mock_attrgetter.return_value = lambda x: 'module'
+        mock_getattr.side_effect = ['provider', None]
 
         # Run
         AnonymizedFaker.check_provider_function('BaseProvider', 'function_name')
 
         # Assert
-        assert mock_getattr.call_args_list[0] == call(mock_faker.providers, 'BaseProvider')
-        assert mock_getattr.call_args_list[1] == call('module', 'function_name')
+        assert mock_attrgetter.call_args_list[0] == call('BaseProvider')
+        assert mock_getattr.call_args_list[0] == call('module', 'function_name')
 
     @patch('rdt.transformers.pii.anonymizer.faker')
     @patch('rdt.transformers.pii.anonymizer.getattr')
-    def test_check_provider_function_other_providers(self, mock_getattr, mock_faker):
+    @patch('rdt.transformers.pii.anonymizer.attrgetter')
+    def test_check_provider_function_other_providers(self, mock_attrgetter, mock_getattr,
+                                                     mock_faker):
         """Test that ``getattr`` is being called with ``provider_name`` and ``function_name``.
 
         Mock:
             - Mock the ``getattr`` from Python to ensure that is being called with the input.
             - Mock faker and ensure that ``getattr`` is being called with ``faker.providers``.
         """
         # Setup
-        mock_getattr.side_effect = ['module', 'provider_class', None]
+        mock_attrgetter.return_value = lambda x: 'module'
+        mock_getattr.side_effect = ['provider_class', None]
 
         # Run
         AnonymizedFaker.check_provider_function('provider_name', 'function_name')
 
         # Assert
-        assert mock_getattr.call_args_list[0] == call(mock_faker.providers, 'provider_name')
-        assert mock_getattr.call_args_list[1] == call('module', 'Provider')
-        assert mock_getattr.call_args_list[2] == call('provider_class', 'function_name')
+        assert mock_attrgetter.call_args_list[0] == call('provider_name')
+        assert mock_getattr.call_args_list[0] == call('module', 'Provider')
+        assert mock_getattr.call_args_list[1] == call('provider_class', 'function_name')
 
     def test_check_provider_function_raise_attribute_error(self):
         """Test that ``check_provider_function`` raises an ``AttributeError``.