def test_load_source_should_strip_smt_pragmas_if_requested(self):
expected_file_content = (
"\n"
"contract C {\n"
"}\n"
)
self.assertEqual(load_source(SMT_SMOKE_TEST_SOL_PATH, SMTUse.STRIP_PRAGMAS), expected_file_content)
def test_load_source_preserves_cr_newlines(self):
expected_output = ("pragma experimental SMTChecker;\r"
"\r"
"contract C {\r"
"}\r")
self.assertEqual(load_source(SMT_CONTRACT_WITH_CR_NEWLINES_SOL_PATH),
expected_output)
def test_load_source_preserves_mixed_newlines(self):
expected_output = (
"\n"
"\n"
"contract C {\r"
"}\r\n"
)
self.assertEqual(load_source(SMT_CONTRACT_WITH_MIXED_NEWLINES_SOL_PATH, SMTUse.STRIP_PRAGMAS), expected_output)
def test_load_source_should_not_strip_smt_pragmas_if_not_requested(self):
self.assertEqual(load_source(SMT_SMOKE_TEST_SOL_PATH, SMTUse.DISABLE),
SMT_SMOKE_TEST_SOL_CODE)
self.assertEqual(load_source(SMT_SMOKE_TEST_SOL_PATH, SMTUse.PRESERVE),
SMT_SMOKE_TEST_SOL_CODE)
def test_load_source(self):
self.assertEqual(load_source(SMT_SMOKE_TEST_SOL_PATH),
SMT_SMOKE_TEST_SOL_CODE)