From 5201db46d20aedf71b21ecd367e0227cf04c8bf4 Mon Sep 17 00:00:00 2001 From: jeremy Date: Mon, 7 Nov 2022 12:15:57 +0100 Subject: [PATCH] Add assert for split_file --- tests/mgbench/datasets.py | 7 +++---- 1 file changed, 3 insertions(+), 4 deletions(-) diff --git a/tests/mgbench/datasets.py b/tests/mgbench/datasets.py index e73169aab..8722ada1d 100644 --- a/tests/mgbench/datasets.py +++ b/tests/mgbench/datasets.py @@ -63,10 +63,9 @@ class Dataset: raise ValueError("The size defined for this variant doesn't " "have the number of vertices and/or edges!") self._num_vertices = self._size["vertices"] self._num_edges = self._size["edges"] - if self.SPLIT_FILES is not None: - self._split_file = self.SPLIT_FILES.get(variant, None) - else: - self._split_file = None + self._split_file = self.SPLIT_FILES.get(variant, None) + assert self._split_file is not None + assert self._split_file != "" def prepare(self, directory): if self._file is not None: