mirror of
https://github.com/NVIDIA-NeMo/DataDesigner
synced 2026-05-24 09:48:29 +00:00
37 lines
1.4 KiB
Python
37 lines
1.4 KiB
Python
# SPDX-FileCopyrightText: Copyright (c) 2025 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
|
|
# SPDX-License-Identifier: Apache-2.0
|
|
|
|
from data_designer.config.utils.code_lang import CodeLang, code_lang_to_syntax_lexer
|
|
|
|
|
|
def test_parse():
|
|
assert CodeLang.parse("ruby") == ("ruby", None)
|
|
assert CodeLang.parse("sql:sqlite") == ("sql", "sqlite")
|
|
assert CodeLang.parse(CodeLang.RUBY) == ("ruby", None)
|
|
assert CodeLang.parse(CodeLang.SQL_SQLITE) == ("sql", "sqlite")
|
|
|
|
|
|
def test_parse_lang():
|
|
assert CodeLang.parse_lang("ruby") == "ruby"
|
|
assert CodeLang.parse_lang(CodeLang.RUBY) == "ruby"
|
|
assert CodeLang.parse_lang("sql:sqlite") == "sql"
|
|
assert CodeLang.parse_lang(CodeLang.SQL_SQLITE) == "sql"
|
|
|
|
|
|
def test_parse_dialect():
|
|
assert CodeLang.parse_dialect("ruby") is None
|
|
assert CodeLang.parse_dialect(CodeLang.RUBY) is None
|
|
assert CodeLang.parse_dialect("sql:sqlite") == "sqlite"
|
|
assert CodeLang.parse_dialect(CodeLang.SQL_SQLITE) == "sqlite"
|
|
|
|
|
|
def test_supported_values():
|
|
assert len(CodeLang.supported_values()) == 16
|
|
|
|
|
|
def test_code_lang_to_syntax_lexer():
|
|
assert code_lang_to_syntax_lexer("ruby") == "ruby"
|
|
assert code_lang_to_syntax_lexer(CodeLang.RUBY) == "ruby"
|
|
assert code_lang_to_syntax_lexer("sql:sqlite") == "sql"
|
|
assert code_lang_to_syntax_lexer(CodeLang.SQL_SQLITE) == "sql"
|
|
assert code_lang_to_syntax_lexer("something-not-in-mapping") == "something-not-in-mapping"
|