use rust_warrior::UnitType;

#[test]
fn test_unit_type_draw() {
    let u = UnitType::Archer;
    assert_eq!(u.draw(), "a");

    let u = UnitType::Captive;
    assert_eq!(u.draw(), "C");

    let u = UnitType::Sludge;
    assert_eq!(u.draw(), "s");

    let u = UnitType::ThickSludge;
    assert_eq!(u.draw(), "S");

    let u = UnitType::Warrior;
    assert_eq!(u.draw(), "@");

    let u = UnitType::Wizard;
    assert_eq!(u.draw(), "w");
}