From 95c69d31605060c937b2ff95d5c27b9462cc091a Mon Sep 17 00:00:00 2001
From: Uttarayan Mondal <uttarayan21@gmail.com>
Date: Sun, 7 Mar 2021 01:45:05 +0530
Subject: [PATCH] Fixed to_text not returning single digits

---
 src/lib.rs  | 42 +++++++++++++++++++++++-------------------
 src/test.rs |  4 ++++
 2 files changed, 27 insertions(+), 19 deletions(-)

diff --git a/src/lib.rs b/src/lib.rs
index 35a5271..c3b3d64 100644
--- a/src/lib.rs
+++ b/src/lib.rs
@@ -79,26 +79,30 @@ pub fn to_text(number: u32) -> String {
         .collect();
     let second_digit_array: [usize; 2] = [0, 3];
     let mut last: Option<u8> = None;
-    for (index, digit) in digits.iter().rev().enumerate() {
-        if second_digit_array.contains(&index) {
-            last = Some(digit.clone());
-            continue;
-        }
-        if let Some(last_digit) = last {
-            let mut buffer = String::new();
-            buffer.push_str(tens_place(digit.clone(), last_digit).unwrap().as_str());
-            if index > 2 {
-                buffer.push_str(place_value(0, index as u8).unwrap().as_str());
+    if digits.len() == 1 {
+        numtext.push_str(digit_to_text(*digits.last().unwrap()).unwrap().as_str());
+    } else {
+        for (index, digit) in digits.iter().rev().enumerate() {
+            if second_digit_array.contains(&index) {
+                last = Some(digit.clone());
+                continue;
+            }
+            if let Some(last_digit) = last {
+                let mut buffer = String::new();
+                buffer.push_str(tens_place(digit.clone(), last_digit).unwrap().as_str());
+                if index > 2 {
+                    buffer.push_str(place_value(0, index as u8).unwrap().as_str());
+                }
+                numtext.insert_str(0, buffer.as_str());
+                last = None;
+            } else {
+                numtext.insert_str(
+                    0,
+                    place_value(digit.clone(), index as u8 + 1)
+                        .unwrap()
+                        .as_str(),
+                )
             }
-            numtext.insert_str(0, buffer.as_str());
-            last = None;
-        } else {
-            numtext.insert_str(
-                0,
-                place_value(digit.clone(), index as u8 + 1)
-                    .unwrap()
-                    .as_str(),
-            )
         }
     }
     numtext
diff --git a/src/test.rs b/src/test.rs
index 1b83856..c056c2d 100644
--- a/src/test.rs
+++ b/src/test.rs
@@ -4,6 +4,9 @@ mod tests {
     fn digits() {
         use crate::digit_to_text;
         assert_eq!(digit_to_text(9).unwrap(), "nine");
+        assert_eq!(digit_to_text(3).unwrap(), "three");
+        assert_eq!(digit_to_text(7).unwrap(), "seven");
+        assert_eq!(digit_to_text(5).unwrap(), "five");
     }
     #[test]
     fn numbers() {
@@ -11,6 +14,7 @@ mod tests {
         assert_eq!(to_text(12345), "twelvethousandthreehundredfortyfive");
         assert_eq!(to_text(81123), "eightyonethousandonehundredtwentythree");
         assert_eq!(to_text(12), "twelve");
+        assert_eq!(to_text(2), "two");
     }
     #[test]
     fn tens() {