Skip to content
Snippets Groups Projects
Commit 2d27975f authored by Raphael Sturgis's avatar Raphael Sturgis
Browse files

fixed engine tests

parent 560e4228
No related branches found
No related tags found
2 merge requests!12version 0.2a,!10Resolve "Image creation bugs with 0 size windows"
This commit is part of merge request !10. Comments created here will be created in the context of that merge request.
Show whitespace changes
Inline Side-by-side
skais/process/data_augmentation/augmentation_engine.py
+ 1
− 1
View file @ 2d27975f
...@@ -28,7 +28,7 @@ class AugmentationEngine: ...@@ -28,7 +28,7 @@ class AugmentationEngine:
self.pipelines.append(Pipeline([translator, flip])) self.pipelines.append(Pipeline([translator, flip]))
def transform(self, x, verbose=0): def transform(self, x, verbose=0):
results = x.copy() results = []
iterator = self.pipelines iterator = self.pipelines
if verbose > 0: if verbose > 0:
......
skais/process/data_augmentation/flip.py
+ 0
− 2
View file @ 2d27975f
...@@ -14,6 +14,4 @@ class Flip(DataTransformer): ...@@ -14,6 +14,4 @@ class Flip(DataTransformer):
df = trajectory.df.copy() df = trajectory.df.copy()
df['latitude'] = -trajectory.df['latitude'] df['latitude'] = -trajectory.df['latitude']
result.append(AISTrajectory(df)) result.append(AISTrajectory(df))
else:
result += x.copy()
return result return result
skais/tests/process/data_augmentation/test_engine.py
+ 131
− 5
View file @ 2d27975f
...@@ -5,7 +5,8 @@ from skais.process.data_augmentation.augmentation_engine import AugmentationEngi ...@@ -5,7 +5,8 @@ from skais.process.data_augmentation.augmentation_engine import AugmentationEngi
import pandas as pd import pandas as pd
class Test_Engine(unittest.TestCase):
class TestEngine(unittest.TestCase):
def setUp(self): def setUp(self):
t1 = AISTrajectory( t1 = AISTrajectory(
pd.DataFrame( pd.DataFrame(
...@@ -28,7 +29,6 @@ class Test_Engine(unittest.TestCase): ...@@ -28,7 +29,6 @@ class Test_Engine(unittest.TestCase):
self.trajectories = [t1, t2] self.trajectories = [t1, t2]
def test_transform_simple_translation(self): def test_transform_simple_translation(self):
engine = AugmentationEngine(translation_values=[(10, 0), (20, 0)], keep_original=False) engine = AugmentationEngine(translation_values=[(10, 0), (20, 0)], keep_original=False)
...@@ -74,20 +74,116 @@ class Test_Engine(unittest.TestCase): ...@@ -74,20 +74,116 @@ class Test_Engine(unittest.TestCase):
) )
expected = [t1, t2, t3, t4] expected = [t1, t2, t3, t4]
self.assertEqual(len(expected), len(result)) self.assertEqual(len(result), len(expected))
for t1, t2 in zip(result, expected): for t1, t2 in zip(result, expected):
pd.testing.assert_frame_equal(t1.df, t2.df) pd.testing.assert_frame_equal(t1.df, t2.df)
def test_transform_simple_flip(self): def test_transform_simple_flip(self):
engine = AugmentationEngine(flip_values=[None, 0], keep_original=False) engine = AugmentationEngine(flip_values=[(None, 0)], keep_original=False)
result = engine.transform(self.trajectories)
t1 = AISTrajectory(
pd.DataFrame(
{
'ts_sec': [i for i in range(10)],
'latitude': [0 for _ in range(10)],
'longitude': [12 + i for i in range(10)]
}
)
)
t2 = AISTrajectory(
pd.DataFrame(
{
'ts_sec': [i for i in range(10)],
'latitude': [12 - i for i in range(10)],
'longitude': [12 + i for i in range(10)]
}
)
)
expected = [t1, t2]
self.assertEqual(len(expected), len(result))
for t1, t2 in zip(result, expected):
pd.testing.assert_frame_equal(t1.df, t2.df)
def test_transform_flip_and_translate(self):
engine = AugmentationEngine(translation_values=[(10, 0)], flip_values=[(None, 0)], keep_original=False)
result = engine.transform(self.trajectories) result = engine.transform(self.trajectories)
t1 = AISTrajectory(
pd.DataFrame(
{
'ts_sec': [i for i in range(10)],
'latitude': [0 for _ in range(10)],
'longitude': [22 + i for i in range(10)]
}
)
)
t2 = AISTrajectory(
pd.DataFrame(
{
'ts_sec': [i for i in range(10)],
'latitude': [-12 + i for i in range(10)],
'longitude': [22 + i for i in range(10)]
}
)
)
t3 = AISTrajectory(
pd.DataFrame(
{
'ts_sec': [i for i in range(10)],
'latitude': [0 for _ in range(10)],
'longitude': [12 + i for i in range(10)]
}
)
)
t4 = AISTrajectory(
pd.DataFrame(
{
'ts_sec': [i for i in range(10)],
'latitude': [12 - i for i in range(10)],
'longitude': [12 + i for i in range(10)]
}
)
)
t5 = AISTrajectory(
pd.DataFrame(
{
'ts_sec': [i for i in range(10)],
'latitude': [0 for _ in range(10)],
'longitude': [22 + i for i in range(10)]
}
)
)
t6 = AISTrajectory(
pd.DataFrame(
{
'ts_sec': [i for i in range(10)],
'latitude': [12 - i for i in range(10)],
'longitude': [22 + i for i in range(10)]
}
)
)
expected = [t1, t2, t3, t4, t5, t6]
self.assertEqual(len(expected), len(result))
for t1, t2 in zip(result, expected):
pd.testing.assert_frame_equal(t1.df, t2.df)
def test_engine_verbose(self):
engine = AugmentationEngine(flip_values=[(None, 0)], keep_original=False)
result = engine.transform(self.trajectories, verbose=1)
t1 = AISTrajectory( t1 = AISTrajectory(
pd.DataFrame( pd.DataFrame(
{ {
'ts_sec': [i for i in range(10)], 'ts_sec': [i for i in range(10)],
'latitude': [0 - i for i in range(10)], 'latitude': [0 for _ in range(10)],
'longitude': [12 + i for i in range(10)] 'longitude': [12 + i for i in range(10)]
} }
) )
...@@ -107,5 +203,35 @@ class Test_Engine(unittest.TestCase): ...@@ -107,5 +203,35 @@ class Test_Engine(unittest.TestCase):
for t1, t2 in zip(result, expected): for t1, t2 in zip(result, expected):
pd.testing.assert_frame_equal(t1.df, t2.df) pd.testing.assert_frame_equal(t1.df, t2.df)
def test_engine_keep_original(self):
engine = AugmentationEngine(flip_values=[(None, 0)], keep_original=True)
result = engine.transform(self.trajectories, verbose=1)
t1 = AISTrajectory(
pd.DataFrame(
{
'ts_sec': [i for i in range(10)],
'latitude': [0 for _ in range(10)],
'longitude': [12 + i for i in range(10)]
}
)
)
t2 = AISTrajectory(
pd.DataFrame(
{
'ts_sec': [i for i in range(10)],
'latitude': [12 - i for i in range(10)],
'longitude': [12 + i for i in range(10)]
}
)
)
expected = self.trajectories + [t1, t2]
self.assertEqual(len(expected), len(result))
for t1, t2 in zip(result, expected):
pd.testing.assert_frame_equal(t1.df, t2.df)
if __name__ == '__main__': if __name__ == '__main__':
unittest.main() unittest.main()
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment