diff --git a/setup.py b/setup.py index caea05b..1b2c748 100644 --- a/setup.py +++ b/setup.py @@ -45,7 +45,6 @@ install_requires=[ "alembic>=0.8.8", "rdflib>=6,<8", - "six>=1.10.0", "SQLAlchemy>=2.0.23", ], entry_points={ diff --git a/test/test_sqlalchemy.py b/test/test_sqlalchemy.py index 7e62b9c..f6cdf27 100644 --- a/test/test_sqlalchemy.py +++ b/test/test_sqlalchemy.py @@ -1,11 +1,5 @@ import unittest - -try: - from unittest.mock import patch, MagicMock -except ImportError: - from mock import patch, MagicMock - -import six +from unittest.mock import patch, MagicMock from rdflib import ( ConjunctiveGraph, @@ -48,7 +42,7 @@ def test_success(self): def test_no_url(self): with patch('rdflib_sqlalchemy.store.sqlalchemy'): - with six.assertRaisesRegex(self, Exception, '.*url.*'): + with self.assertRaisesRegex(Exception, '.*url.*'): self.graph.open({'random_key': 'something'}, create=True)