Chromium Code Reviews| OLD | NEW |
|---|---|
| 1 #!/usr/bin/env python | 1 #!/usr/bin/env python |
| 2 # Copyright (c) 2013 The Chromium Authors. All rights reserved. | 2 # Copyright (c) 2013 The Chromium Authors. All rights reserved. |
| 3 # Use of this source code is governed by a BSD-style license that can be | 3 # Use of this source code is governed by a BSD-style license that can be |
| 4 # found in the LICENSE file. | 4 # found in the LICENSE file. |
| 5 | 5 |
| 6 # Self-test for skimage. | 6 # Self-test for skimage. |
| 7 | 7 |
| 8 import filecmp | 8 import filecmp |
| 9 import os | 9 import os |
| 10 import subprocess | 10 import subprocess |
| 11 import sys | 11 import sys |
| 12 import tempfile | |
| 12 | 13 |
| 13 class BinaryNotFoundException(Exception): | 14 class BinaryNotFoundException(Exception): |
| 14 def __str__ (self): | 15 def __str__ (self): |
| 15 return ("Could not find binary!\n" | 16 return ("Could not find binary!\n" |
| 16 "Did you forget to build the tools project?\n" | 17 "Did you forget to build the tools project?\n" |
| 17 "Self tests failed") | 18 "Self tests failed") |
| 18 | 19 |
| 19 # Find a path to the binary to use. Iterates through a list of possible | 20 # Find a path to the binary to use. Iterates through a list of possible |
| 20 # locations the binary may be. | 21 # locations the binary may be. |
| 21 def PickBinaryPath(base_dir): | 22 def PickBinaryPath(base_dir): |
| (...skipping 22 matching lines...) Expand all Loading... | |
| 44 | 45 |
| 45 trunk_dir = os.path.normpath(os.path.join(file_dir, os.pardir, os.pardir)) | 46 trunk_dir = os.path.normpath(os.path.join(file_dir, os.pardir, os.pardir)) |
| 46 | 47 |
| 47 # Find the binary | 48 # Find the binary |
| 48 skimage_binary = PickBinaryPath(trunk_dir) | 49 skimage_binary = PickBinaryPath(trunk_dir) |
| 49 print "Running " + skimage_binary | 50 print "Running " + skimage_binary |
| 50 | 51 |
| 51 # Generate an expectations file from known images. | 52 # Generate an expectations file from known images. |
| 52 images_dir = os.path.join(file_dir, "skimage", "input", | 53 images_dir = os.path.join(file_dir, "skimage", "input", |
| 53 "images-with-known-hashes") | 54 "images-with-known-hashes") |
| 54 expectations_path = os.path.join(file_dir, "skimage", "output-actual", | 55 expectations_path = os.path.join(file_dir, "skimage", "output-actual", |
|
scroggo
2013/06/19 20:34:16
Patch set 4 reverts the changes in patch set 3.
I
| |
| 55 "create-expectations", "expectations.json") | 56 "create-expectations", "expectations.json") |
| 56 subprocess.check_call([skimage_binary, "--readPath", images_dir, | 57 subprocess.check_call([skimage_binary, "--readPath", images_dir, |
| 57 "--createExpectationsPath", expectations_path]) | 58 "--createExpectationsPath", expectations_path]) |
| 58 | 59 |
| 59 # Make sure the expectations file was generated correctly. | 60 # Make sure the expectations file was generated correctly. |
| 60 golden_expectations = os.path.join(file_dir, "skimage", "output-expected", | 61 golden_expectations = os.path.join(file_dir, "skimage", "output-expected", |
| 61 "create-expectations", | 62 "create-expectations", |
| 62 "expectations.json") | 63 "expectations.json") |
| 63 DieIfFilesMismatch(expected=golden_expectations, actual=expectations_path) | 64 DieIfFilesMismatch(expected=golden_expectations, actual=expectations_path) |
| 64 | 65 |
| 65 # Tell skimage to read back the expectations file it just wrote, and | 66 # Tell skimage to read back the expectations file it just wrote, and |
| 66 # confirm that the images in images_dir match it. | 67 # confirm that the images in images_dir match it. |
| 67 subprocess.check_call([skimage_binary, "--readPath", images_dir, | 68 subprocess.check_call([skimage_binary, "--readPath", images_dir, |
| 68 "--readExpectationsPath", expectations_path]) | 69 "--readExpectationsPath", expectations_path]) |
| 69 | 70 |
| 70 # TODO(scroggo): Add a test that compares expectations and image files that | 71 # TODO(scroggo): Add a test that compares expectations and image files that |
| 71 # are known to NOT match, and make sure it returns an error. | 72 # are known to NOT match, and make sure it returns an error. |
| 72 | 73 |
| 74 # Generate an expectations file from an empty directory. | |
| 75 empty_dir = tempfile.mkdtemp() | |
| 76 expectations_path = os.path.join(file_dir, "skimage", "output-actual", | |
| 77 "empty-dir", "expectations.json") | |
| 78 subprocess.check_call([skimage_binary, "--readPath", empty_dir, | |
| 79 "--createExpectationsPath", expectations_path]) | |
| 80 golden_expectations = os.path.join(file_dir, "skimage", "output-expected", | |
| 81 "empty-dir", "expectations.json") | |
| 82 DieIfFilesMismatch(expected=golden_expectations, actual=expectations_path) | |
| 83 os.rmdir(empty_dir) | |
| 84 | |
| 85 # Generate an expectations file from a nonexistent directory. | |
| 86 expectations_path = os.path.join(file_dir, "skimage", "output-actual", | |
| 87 "nonexistent-dir", "expectations.json") | |
| 88 subprocess.check_call([skimage_binary, "--readPath", "/nonexistent/dir", | |
| 89 "--createExpectationsPath", expectations_path]) | |
| 90 golden_expectations = os.path.join(file_dir, "skimage", "output-expected", | |
| 91 "nonexistent-dir", "expectations.json") | |
| 92 DieIfFilesMismatch(expected=golden_expectations, actual=expectations_path) | |
| 93 | |
| 73 # Done with all tests. | 94 # Done with all tests. |
| 74 print "Self tests succeeded!" | 95 print "Self tests succeeded!" |
| 75 | 96 |
| 76 if __name__ == "__main__": | 97 if __name__ == "__main__": |
| 77 main() | 98 main() |
| OLD | NEW |