Fix unreliable test that assumes current file will always be named .py.

1 files changed, 5 insertions, 2 deletions

diff --git a/license_protected_downloads/tests/test_buildinfo.py b/license_protected_downloads/tests/test_buildinfo.py
index 99005b4..f315529 100644
--- a/license_protected_downloads/tests/test_buildinfo.py
+++ b/license_protected_downloads/tests/test_buildinfo.py
@@ -231,11 +231,14 @@ class BuildInfoTests(unittest.TestCase):
         build_info.full_file_name = file_path
         build_info.build_info_array = [{}]
         build_info.file_info_array = [{}]
-        data = ["Format-Version: 2.0", "Files-Pattern: *.py", "License-Type: protected"]
+        data = ["Format-Version: 2.0",
+                "Files-Pattern: *.py*",
+                "License-Type: protected"]
         build_info.parseData(data)
         build_info.file_info_array = build_info.getInfoForFile()
 
-        self.assertEquals(build_info.file_info_array, [{'license-type': 'protected'}])
+        self.assertEquals(build_info.file_info_array,
+                          [{'license-type': 'protected'}])
 
     def test_remove_false_positives_no_blocks_in_array(self):
         file_path = os.path.abspath(__file__)