Function remove_optimizer_rule

datafusion/core/src/execution/context/mod.rs:2888–2930

()

    #[tokio::test]
    async fn remove_optimizer_rule() -> Result<()> {
        let get_optimizer_rules = |ctx: &SessionContext| {
            ctx.state()
                .optimizer()
                .rules
                .iter()
                .map(|r| r.name().to_owned())
                .collect::<HashSet<_>>()
        };

        let ctx = SessionContext::new();
        assert!(get_optimizer_rules(&ctx).contains("simplify_expressions"));

        // default plan
        let plan = ctx
            .sql("select 1 + 1")
            .await?
            .into_optimized_plan()?
            .to_string();
        assert_snapshot!(plan, @r"
        Projection: Int64(2) AS Int64(1) + Int64(1)
          EmptyRelation: rows=1
        ");

        assert!(ctx.remove_optimizer_rule("simplify_expressions"));
        assert!(!get_optimizer_rules(&ctx).contains("simplify_expressions"));

        // plan without the simplify_expressions rule
        let plan = ctx
            .sql("select 1 + 1")
            .await?
            .into_optimized_plan()?
            .to_string();
        assert_snapshot!(plan, @r"
        Projection: Int64(1) + Int64(1)
          EmptyRelation: rows=1
        ");

        // attempting to remove a non-existing rule returns false
        assert!(!ctx.remove_optimizer_rule("simplify_expressions"));

        Ok(())
    }

    #[test]
    fn test_parse_duration() {