Exemplos de Metadata._get_pii_fields em Python

Exemplo n.º 1

    def test__get_pii_fields(self):
        """Test get pii fields"""
        # Setup
        table_meta = {
            'fields': {
                'foo': {
                    'type': 'categorical',
                    'pii': True,
                    'pii_category': 'email'
                },
                'bar': {
                    'type': 'categorical',
                    'pii_category': 'email'
                }
            }
        }

        # Run
        metadata = Mock(spec=Metadata)
        metadata.get_table_meta.return_value = table_meta

        table_name = 'test'

        result = Metadata._get_pii_fields(metadata, table_name)

        # Asserts
        expected = {'foo': 'email'}

        assert result == expected

Exemplo n.º 2

Arquivo: test_metadata.py Projeto: glf818/SDV

    def test__get_pii_fields(self):
        """Test get pii fields"""
        # Setup
        table_meta = {
            'fields': {
                'foo': {
                    'type': 'categorical',
                    'pii': True,
                    'pii_category': 'email'
                },
                'bar': {
                    'type': 'categorical',
                    'pii_category': 'email'
                }
            }
        }
        metadata = Mock(spec_set=Metadata)
        metadata.get_table_meta.return_value = table_meta

        # Run
        result = Metadata._get_pii_fields(metadata, 'test')

        # Asserts
        assert result == {'foo': 'email'}