Chore: Code cleanup

msiemens · Feb 25, 2019 · 74b89fa · 74b89fa
1 parent a320136
commit 74b89fa
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 3 deletions.
diff --git a/tests/test_tinydb.py b/tests/test_tinydb.py
@@ -5,8 +5,8 @@
 import pytest
 
 from tinydb import TinyDB, where
-from tinydb.storages import MemoryStorage
 from tinydb.middlewares import Middleware
+from tinydb.storages import MemoryStorage
 
 
 def test_purge(db):

diff --git a/tinydb/database.py b/tinydb/database.py
@@ -170,7 +170,8 @@ def __repr__(self):
             'tables_count={}'.format(len(self.tables())),
             'default_table_documents_count={}'.format(self.__len__()),
             'all_tables_documents_count={}'.format(
-                ['{}={}'.format(table, len(self.table(table))) for table in self.tables()]),
+                ['{}={}'.format(table, len(self.table(table)))
+                 for table in self.tables()]),
         ]
 
         return '<{} {}>'.format(type(self).__name__, ', '.join(args))
@@ -406,7 +407,7 @@ def _write(self, values):
         :type values: DataProxy | dict
         """
 
-        self._query_cache.clear()
+        self.clear_cache()
         self._storage.write(values)
 
     def __len__(self):