Function test_extend_entity_inference_parser

src/cosql/inference/extend_entity.rs:30–63

()

    #[test]
    fn test_extend_entity_inference_parser() {
        let values = [
            (
                "extend $person1 (
                    salary: 100000
                )",
                ExtendEntityInference {
                    variable: "person1".to_string(),
                    attributes: vec![Attribute {
                        name: "salary".to_string(),
                        value: Value::Int(100000),
                    }],
                },
            ),
            (
                "extend $city1 (
                    visitors: $visitors
                )",
                ExtendEntityInference {
                    variable: "city1".to_string(),
                    attributes: vec![Attribute {
                        name: "visitors".to_string(),
                        value: Value::Variable("visitors".to_string()),
                    }],
                },
            ),
        ];

        for (source, expected) in values {
            let (_, parsed) = parse_extend_entity_inference(source).unwrap();

            assert_eq!(parsed, expected);
        }
    }
}