def _check_basic(self, input_module_path, input_symbol, input_statement):
"""Checks all but to_str(), returning the imported_symbol instance."""
statement = ast.parse(input_statement).body[0]
symbol = ImportedSymbol.get_symbol_from_statement(statement)
msg = f'Expected symbol name "{input_symbol}", found "{symbol}"'
self.assertEqual(symbol, input_symbol, msg)
module_path = ImportedSymbol.get_module_path_from_statement(statement)
msg = f'Expected module path "{input_module_path}", ' f'found "{module_path}"'
self.assertEqual(module_path, input_module_path, msg)
imported_symbol = ImportedSymbol(module_path, symbol)
self.assertEqual(imported_symbol,
ImportedSymbol.from_statement(statement))
return imported_symbol
def test_incorrect_statement_type(self):
statement = ast.parse("pass").body[0]
with self.assertRaises(ValueError):
_ = ImportedSymbol.get_symbol_from_statement(statement)
