From 6de215e64ec4655e6bd644c1a214482fee8bbc77 Mon Sep 17 00:00:00 2001
From: Andreas Nordin <anordin@n42.se>
Date: Mon, 29 Jun 2020 14:24:59 +0200
Subject: [PATCH] Fix AWG string casing in output

Convert e.g. 'awg, 'AwG' to upper case for consistent rendering. Leave
any other input gauge units as they were.
---
 src/wireviz/DataClasses.py | 7 ++++++-
 1 file changed, 6 insertions(+), 1 deletion(-)

diff --git a/src/wireviz/DataClasses.py b/src/wireviz/DataClasses.py
index a2f2a9e..6bc9eb8 100644
--- a/src/wireviz/DataClasses.py
+++ b/src/wireviz/DataClasses.py
@@ -83,7 +83,12 @@ class Cable:
             except Exception:
                 raise Exception('Gauge must be a number, or number and unit separated by a space')
             self.gauge = g
-            self.gauge_unit = u.replace('mm2', 'mm\u00B2')
+
+            if u.upper() == 'AWG':
+                self.gauge_unit = u.upper()
+            else:
+                self.gauge_unit = u.replace('mm2', 'mm\u00B2')
+
         elif self.gauge is not None:  # gauge specified, assume mm2
             if self.gauge_unit is None:
                 self.gauge_unit = 'mm\u00B2'