()
| 514 | |
| 515 | #[test] |
| 516 | fn test_extract_class_with_bases() { |
| 517 | let mut parser = PythonParser::new(); |
| 518 | let source = r#" |
| 519 | class Admin(User, Serializable): |
| 520 | pass |
| 521 | "#; |
| 522 | let entities = parser.extract(source, "models.py"); |
| 523 | let admin = entities.iter().find(|e| e.name == "Admin").unwrap(); |
| 524 | assert!( |
| 525 | admin.signature.as_ref().unwrap().contains("User"), |
| 526 | "Signature should include base classes: {:?}", |
| 527 | admin.signature |
| 528 | ); |
| 529 | } |
| 530 | |
| 531 | #[test] |
| 532 | fn test_extract_module_constants() { |