def test_predict():
dd = DGADetector()
test_domains = cudf.Series(["nvidia.com", "dfsdfsdf"])
dd.load_model(model_filepath)
actual_output = dd.predict(test_domains)
expected_output = cudf.Series([1, 0])
assert actual_output.equals(actual_output)
def test_predict():
if torch.cuda.is_available():
dd = DGADetector()
test_domains = cudf.Series(["nvidia.com", "dfsdfsdf"])
dd.load_model(MODEL_FILENAME)
actual_output = dd.predict(test_domains)
expected_output = cudf.Series([1, 0])
assert actual_output.equals(expected_output)