Skip to content

Commit 17c9306

Browse files
committed
[rdf_converter.py] Adjust Writer Test for RDF Formats
Check for RDF format extensions given in format dictionary.
1 parent a9ee81b commit 17c9306

File tree

1 file changed

+4
-4
lines changed

1 file changed

+4
-4
lines changed

odml/tools/rdf_converter.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -212,10 +212,10 @@ def get_rdf_str(self, rdf_format="turtle"):
212212
def write_file(self, filename, rdf_format="turtle"):
213213
data = self.get_rdf_str(rdf_format)
214214
filename_ext = filename
215-
if not filename.find("." + rdf_format):
216-
filename_ext += "." + rdf_format
217-
with open(filename_ext, "w") as file:
218-
file.write(data)
215+
if filename.find(self._conversion_formats.get(rdf_format)) < 0:
216+
filename_ext += self._conversion_formats.get(rdf_format)
217+
with open(filename_ext, "w") as wFile:
218+
wFile.write(data)
219219

220220

221221
class RDFReader(object):

0 commit comments

Comments
 (0)