From f41a0a10eecb443c87d706e80398d578defeda53 Mon Sep 17 00:00:00 2001
From: Robin Thomas <rthomas900@gmail.com>
Date: Sat, 1 Apr 2017 23:45:00 -0400
Subject: [PATCH] complete basic write test

---
 tests/gspread_dataframe_test.py | 23 ++++++++++++-----------
 tests/mock_worksheet.py         |  3 ++-
 2 files changed, 14 insertions(+), 12 deletions(-)

diff --git a/tests/gspread_dataframe_test.py b/tests/gspread_dataframe_test.py
index ee0f93d..8337a00 100644
--- a/tests/gspread_dataframe_test.py
+++ b/tests/gspread_dataframe_test.py
@@ -4,6 +4,7 @@
 from gspread_dataframe import *
 import numpy as np
 import pandas as pd
+from difflib import SequenceMatcher
 
 import unittest
 from unittest.mock import Mock, MagicMock
@@ -13,15 +14,15 @@
 # Expected results
 
 COLUMN_NAMES = [
-    'Thingy', 
-    'Syntax', 
-    'Numeric Column', 
-    'Formula Column', 
-    'Date Column', 
-    'Values are...', 
-    'Selection', 
-    'Label(s) referencible in chart title', 
-    'Dialect-specific implementations', 
+    'Thingy',
+    'Syntax',
+    'Numeric Column',
+    'Formula Column',
+    'Date Column',
+    'Values are...',
+    'Selection',
+    'Label(s) referencible in chart title',
+    'Dialect-specific implementations',
     'Notes'
 ]
 
@@ -137,5 +138,5 @@ def test_write_basic(self):
         df = get_as_dataframe(self.sheet)
         set_with_dataframe(self.sheet, df, resize=True)
         self.sheet.resize.assert_called_once_with(10, 10)
-        from difflib import SequenceMatcher
-        #self.sheet.client.post_cells.assert_called_once_with(self.sheet, POST_CELLS_EXPECTED.encode('utf8'))
+        self.sheet.client.post_cells.assert_called_once()
+        self.sheet.client.post_cells.assert_called_once_with(self.sheet, POST_CELLS_EXPECTED)
diff --git a/tests/mock_worksheet.py b/tests/mock_worksheet.py
index db73e9e..ec69b5f 100644
--- a/tests/mock_worksheet.py
+++ b/tests/mock_worksheet.py
@@ -13,9 +13,10 @@ def contents_of_file(filename, et_parse=True):
 
 CELL_FEED = contents_of_file('cell_feed.xml')
 WORKSHEET_FEED = contents_of_file('worksheet_feed.xml')
-POST_CELLS_EXPECTED = contents_of_file('post_cells_expected.xml', False)
+POST_CELLS_EXPECTED = contents_of_file('post_cells_expected.xml', False).encode('utf8')
 
 class MockWorksheet(object):
+
     def _fetch_cells(self):
         return [Cell(self, elem) for elem in CELL_FEED.findall(_ns('entry'))]