def test_get_non_existing_file_format_raises(data_file: Text): with pytest.raises(AttributeError): get_file_format_extension(data_file)
def test_get_supported_file_format(data_file: Text, expected_format: Text): fformat = get_file_format_extension(data_file) assert fformat == expected_format