def setUp(self):
self.traj = Trajectory([1, 2, 3, 5], [0, 2, 2, 4], [
"2000-01-01 00:01:00", "2000-02-01 00:01:00",
"2000-03-01 00:01:00", "2000-04-01 00:01:00"
])
self.traj2 = Trajectory([0, 2, 0], [0, 2, 4], [
"2000-01-01 00:01:00", "2000-02-01 00:01:00", "2000-03-01 00:01:00"
])
def test_tamanho_dos_tres_atributos_sao_diferentes(self):
try:
Trajectory([5, 5, 4, 5], [4, 5],
['2000-05-01', '2000-05-01', '2000-05-01'])
assert (False)
except:
assert (True)
def test_criar_objeto_de_dois_pontos(self):
try:
Trajectory([1, 2], [0, 2], ['2000-01-01', '2000-02-01'])
assert (True)
except:
assert (False)
pass
def test_tamanho_de_y_e_diferente_de_t_e_x(self):
try:
Trajectory(
[5, 5, 4, 5], [4, 5],
['2000-04-01', '2000-05-01', '2000-05-01', '2000-05-01'])
assert (False)
except:
assert (True)
def test_criar_obj_de_tres_pontos(self):
try:
Trajectory([1, 2, 3], [0, 2, 3],
['2000-01-01', '2000-02-01', '2000-03-01'])
assert (True)
except:
assert (False)
pass
def test_criar_obj_de_dez_pontos(self):
try:
Trajectory([1, 2, 2, 3, 3, 4, 5, 5, 3, 3],
[0, 2, 4, 4, 2, 2, 2, 1, 1, 2], [
"2000-01-01 00:01:00", "2000-02-01 00:01:00",
"2000-03-01 00:01:00", "2000-04-01 00:01:00",
"2000-05-01 00:01:00", "2000-06-01 00:01:00",
"2000-07-01 00:01:00", "2000-08-01 00:01:00",
"2000-09-01 00:01:00", "2000-10-01 00:01:00"
])
assert (True)
except:
assert (False)
pass
def test_intersection_com_mock(self):
ls = LineString([(1.5, 1, 9.48024060e+08), (2, 2, 9.49363260e+08),
(3, 2, 9.51868860e+08), (4, 3, 9.53208060e+08)])
poly = Polygon([(1, 1), (1, 3), (4, 3), (4, 1), (1, 1)])
self.traj.intersection_shapely = MagicMock(return_value=ls)
response = self.traj.intersection_shapely(poly)
ls = np.array(ls)
trajMock = self.traj.to_Trajectory(response)
traj = Trajectory(ls[:, 0], ls[:, 1], ls[:, 2])
assert (np.array_equal(trajMock.getX(), traj.getX()))
assert (np.array_equal(trajMock.getY(), traj.getY()))
assert (np.array_equal(trajMock.getTime(), traj.getTime()))