|
|
|
@ -29,11 +29,26 @@ end"""
|
|
|
|
|
self.assertTrue(LuaSegmenter("local a").is_valid())
|
|
|
|
|
self.assertFalse(LuaSegmenter("a b c 1 2 3").is_valid())
|
|
|
|
|
|
|
|
|
|
# TODO: Investigate flakey-ness.
|
|
|
|
|
@pytest.mark.skip(
|
|
|
|
|
reason=(
|
|
|
|
|
"Flakey. To be investigated. See "
|
|
|
|
|
"https://github.com/langchain-ai/langchain/actions/runs/7907779756/job/21585580650." # noqa: E501
|
|
|
|
|
)
|
|
|
|
|
)
|
|
|
|
|
def test_extract_functions_classes(self) -> None:
|
|
|
|
|
segmenter = LuaSegmenter(self.example_code)
|
|
|
|
|
extracted_code = segmenter.extract_functions_classes()
|
|
|
|
|
self.assertEqual(extracted_code, self.expected_extracted_code)
|
|
|
|
|
|
|
|
|
|
# TODO: Investigate flakey-ness.
|
|
|
|
|
@pytest.mark.skip(
|
|
|
|
|
reason=(
|
|
|
|
|
"Flakey. To be investigated. See "
|
|
|
|
|
"https://github.com/langchain-ai/langchain/actions/runs/7923203031/job/21632416298?pr=17599 " # noqa: E501
|
|
|
|
|
"and https://github.com/langchain-ai/langchain/actions/runs/7923784089/job/2163420864." # noqa: E501
|
|
|
|
|
)
|
|
|
|
|
)
|
|
|
|
|
def test_simplify_code(self) -> None:
|
|
|
|
|
segmenter = LuaSegmenter(self.example_code)
|
|
|
|
|
simplified_code = segmenter.simplify_code()
|
|
|
|
|