format pis #206 by patricia-salles · Pull Request #224 · brazilian-utils/python
| def test_format_pis(self): | |
| with patch("brutils.formatpis.is_valid", return_value=True) as mock_is_valid: | |
| # When cpf is_valid, returns formatted cpf | |
| self.assertEqual(format_pis("14372195539"), "143.72195.53-9") | |
| # Checks if function is_valid_cpf is called | |
| mock_is_valid.assert_called_once_with("14372195539") | |
| with patch("brutils.formatpis.is_valid", return_value=False) as mock_is_valid: | |
| # When cpf isn't valid, returns None | |
| self.assertIsNone(format_pis("14372195539")) | |
| def test_format_pis(self): | |
| with patch("brutils.pis.is_valid", return_value=True) as mock_is_valid: | |
| # When PIS is_valid, returns formatted PIS | |
| self.assertEqual(format_pis("14372195539"), "143.72195.53-9") | |
| # Checks if function is_valid_pis is called | |
| mock_is_valid.assert_called_once_with("14372195539") | |
| with patch("brutils.pis.is_valid", return_value=False) as mock_is_valid: | |
| # When PIS isn't valid, returns None | |
| self.assertIsNone(format_pis("14372195539")) |