fix set_package function

Author: MrMaydo

src/header_generator.py

@@ -2,7 +2,7 @@
 
 
 def set_package(package: str) -> str:
-    PACKAGE_REGEX = r"^(?:[a-z_][a-z0-9_]*)(?:\.(?:[a-z_][a-z0-9_]*))*$"
+    PACKAGE_REGEX = r"^(?:[A-Za-z_][A-Za-z0-9_]*)(?:\.(?:[A-Za-z_][A-Za-z0-9_]*))*$"
     if not re.match(PACKAGE_REGEX, package):
         raise ValueError(f"Invalid package: '{package}'")
-    return f"package {package}"
+    return f"package {package};"

tests/test_header_generator.py

@@ -5,13 +5,17 @@
 
 def test_set_package():
     example = "org.example.hyphenated_name"
-    expected = f"package {example}"
+    expected = f"package {example};"
     assert set_package(example) == expected
 
     example_2 = "com.example._123name"
-    expected_2 = f"package {example_2}"
+    expected_2 = f"package {example_2};"
     assert set_package(example_2) == expected_2
 
+    example_3 = "com.exAmple.Name"
+    expected_3 = f"package {example_3};"
+    assert set_package(example_3) == expected_3
+
 
 def test_set_package_illegal_name():
     example = "org.example.hyphenated-name"