Skip to content
GitLab
Menu
Projects
Groups
Snippets
/
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in
Toggle navigation
Menu
Open sidebar
Christian Fischer Pedersen
AIR
Commits
810e5e62
Commit
810e5e62
authored
Dec 16, 2021
by
thecml
Browse files
removed old test files
parent
4350076b
Changes
1
Hide whitespace changes
Inline
Side-by-side
ml/tests/test_cleaner.py
View file @
810e5e62
import
pandas
as
pd
import
datetime
from
tools.cleaner
import
Cleaner202
0
from
tools.cleaner
import
Cleaner202
1
def
test_filter_ats_on_ids
():
df
=
pd
.
DataFrame
([
i
for
i
in
range
(
10
)],
columns
=
[
'CitizenId'
])
ids
=
[
i
for
i
in
range
(
5
)]
filtered_list
=
Cleaner202
0
().
filter_ats_on_ids
(
df
,
ids
)
filtered_list
=
Cleaner202
1
().
filter_ats_on_ids
(
df
,
ids
)
assert
not
filtered_list
.
empty
assert
len
(
filtered_list
)
==
5
...
...
@@ -57,7 +57,7 @@ def test_remove_tainted_histories():
def
test_remove_deprecated_device_data
():
df
=
pd
.
DataFrame
({
'DevHMINumber'
:
[
'899,997'
,
'899,989'
,
'899,999'
]})
res
=
Cleaner202
0
().
remove_deprecated_device_data
(
df
)
res
=
Cleaner202
1
().
remove_deprecated_device_data
(
df
)
assert
not
res
.
empty
assert
len
(
res
)
==
2
...
...
Write
Preview
Supports
Markdown
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment