def get_condition_rhs_num(if_node): assert isinstance(if_node, ast.If) assert isinstance(if_node.test, ast.Compare) assert isinstance(if_node.test.comparators[0], ast.Num) return if_node.test.comparators[0].n