|
3 | 3 |
|
4 | 4 | from nose.tools import assert_true, assert_equal |
5 | 5 | from numpy.testing import assert_array_equal |
6 | | -from datetime import datetime |
7 | 6 | from smartdispatch import utils |
8 | 7 |
|
9 | 8 |
|
@@ -33,38 +32,6 @@ def test_generate_name_from_command(): |
33 | 32 | assert_equal(smartdispatch.generate_name_from_command(command)[date_lenght:], expected) |
34 | 33 |
|
35 | 34 |
|
36 | | -def test_generate_name_from_arguments(): |
37 | | - prefix = "prefix_" |
38 | | - |
39 | | - arguments = [["my_command"], ["args1a", "args1b", "args1c"], ["args2a", "args2b"]] |
40 | | - expected = prefix + "my_command_args1a-args1c_args2a-args2b" |
41 | | - assert_equal(smartdispatch.generate_name_from_arguments(arguments, prefix=prefix), expected) |
42 | | - |
43 | | - max_length_arg = 7 |
44 | | - arguments = [["command"], ["verylongargs1a", "verylongargs1b", "verylongargs1c"], ["args2a", "args2b"]] |
45 | | - expected = prefix + "command_" + arguments[1][0][-max_length_arg:] + "-" + arguments[1][-1][-max_length_arg:] + "_args2a-args2b" |
46 | | - assert_equal(smartdispatch.generate_name_from_arguments(arguments, max_length_arg, prefix=prefix), expected) |
47 | | - |
48 | | - max_length = 23 |
49 | | - arguments = [["command"], ["verylongargs1a", "verylongargs1b", "verylongargs1c"], ["args2a", "args2b"]] |
50 | | - expected = "command_" + arguments[1][0] + "-" + arguments[1][-1] + "_args2a-args2b" |
51 | | - expected = prefix + expected[-max_length:] |
52 | | - assert_equal(smartdispatch.generate_name_from_arguments(arguments, max_length=max_length, prefix=prefix), expected) |
53 | | - |
54 | | - # Test path arguments in command |
55 | | - arguments = [["command"], ["path/argument/1", "path/argument/2", "path/argument/3"]] |
56 | | - expected = prefix + "command_pathargument1-pathargument3" |
57 | | - assert_equal(smartdispatch.generate_name_from_arguments(arguments, prefix=prefix), expected) |
58 | | - |
59 | | - # Make sure default prefix does not raise exception |
60 | | - arguments = [["command"]] |
61 | | - results = smartdispatch.generate_name_from_arguments(arguments) |
62 | | - expect_datetime = datetime.now() |
63 | | - assert_equal(results.split("_")[-1], arguments[0][0]) |
64 | | - result_datetime = datetime.strptime("_".join(results.split("_")[:-1]), '%Y-%m-%d_%H-%M-%S') |
65 | | - assert_true(result_datetime <= expect_datetime) |
66 | | - |
67 | | - |
68 | 35 | def test_get_commands_from_file(): |
69 | 36 | commands = ["command1 arg1 arg2", |
70 | 37 | "command2", |
@@ -158,7 +125,7 @@ def test_replace_uid_tag(): |
158 | 125 |
|
159 | 126 | commands = ["a command with a {UID} tag"] * 10 |
160 | 127 | uid = utils.generate_uid_from_string(commands[0]) |
161 | | - assert_array_equal(smartdispatch.replace_uid_tag(commands), [commands[0].replace("{UID}", uid)]*len(commands)) |
| 128 | + assert_array_equal(smartdispatch.replace_uid_tag(commands), [commands[0].replace("{UID}", uid)] * len(commands)) |
162 | 129 |
|
163 | 130 |
|
164 | 131 | def test_get_available_queues(): |
|
0 commit comments