1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
|
/*
* Copyright (C) 2020 The Android Open Source Project
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
package com.android.timezone.location.data_pipeline.steps.tzs2polygons_tzs2cellunions;
import static com.android.timezone.location.data_pipeline.steps.TestSupport.copyTestResource;
import static org.junit.Assert.assertEquals;
import static org.junit.Assert.assertTrue;
import com.android.timezone.location.data_pipeline.steps.TestSupport;
import com.android.timezone.location.data_pipeline.steps.Types.ProtoStorageFormat;
import com.android.timezone.location.data_pipeline.steps.Types.TzS2CellUnion;
import com.android.timezone.location.data_pipeline.steps.TzIds;
import com.android.timezone.location.data_pipeline.steps.TzS2PolygonsToTzS2CellUnions;
import org.junit.After;
import org.junit.Before;
import org.junit.Test;
import java.io.File;
import java.io.IOException;
import java.nio.file.Path;
/** Tests for {@link TzS2PolygonsToTzS2CellUnions}. */
public final class TzS2PolygonsToTzS2CellUnionsTest {
private Path mTempDirPath;
@Before
public void setUp() throws IOException {
mTempDirPath = TestSupport.createTempDir(this.getClass());
}
@After
public void tearDown() throws IOException {
TestSupport.deleteDirectory(mTempDirPath);
}
/**
* A regression test that should spot obvious, possibly unintentional, changes to the output
* data by executing the test class with known input and output data.
*/
@Test
public void regressionTest() throws Exception {
// A file containing just the polygons for Europe/London.
String inputResource = "Pacific_Pitcairn_tzs2polygons.prototxt";
String timeZoneId = "Pacific/Pitcairn";
ProtoStorageFormat storageFormat = ProtoStorageFormat.TEXT;
Path inputDirPath = mTempDirPath.resolve("input");
copyTestResource(getClass(), inputResource, inputDirPath);
final int threads = 1;
Path outputDirPath = mTempDirPath.resolve("output");
final int s2Level = 12;
String[] args = {
"--input", inputDirPath.toAbsolutePath().toString(),
"--num-threads", Integer.toString(threads),
"--output", outputDirPath.toAbsolutePath().toString(),
"--max-s2-level", Integer.toString(s2Level),
};
TzS2PolygonsToTzS2CellUnions.main(args);
Path expectedOutputDirPath = mTempDirPath.resolve("expected");
Path expectedOutputFilePath = copyTestResource(getClass(),
TzIds.fileNameSafe(timeZoneId) + TzS2CellUnion.getFileSuffix(storageFormat),
expectedOutputDirPath
);
String expectedFileName = expectedOutputFilePath.getFileName().toString();
File actualOutputFile = new File(outputDirPath.toFile(), expectedFileName);
assertTrue("Expecting " + actualOutputFile + " to exist", actualOutputFile.exists());
TzS2CellUnion actual = TzS2CellUnion.load(actualOutputFile, storageFormat);
TzS2CellUnion expected = TzS2CellUnion.load(expectedOutputFilePath.toFile(), storageFormat);
assertEquals(expected, actual);
}
}
|
