UT: Fix imports in test_scan_target_generator

2021-12-16 08:33:38 -05:00 · 2021-12-16 08:33:38 -05:00 · db246d6740
parent 7c786b0883
commit db246d6740
1 changed files with 2 additions and 1 deletions
--- a/monkey/tests/unit_tests/infection_monkey/network/test_scan_target_generator.py
+++ b/monkey/tests/unit_tests/infection_monkey/network/test_scan_target_generator.py
@ -1,11 +1,12 @@
 from itertools import chain

 import pytest
-from network.scan_target_generator import NetworkAddress, _filter_invalid_ranges

 from common.network.network_range import InvalidNetworkRangeError
 from infection_monkey.network.scan_target_generator import (
+    NetworkAddress,
    NetworkInterface,
+    _filter_invalid_ranges,
    compile_scan_target_list,
 )