specify schema for reading

universalcore · Feb 13, 2015 · fe7942e · fe7942e
1 parent 5dfe840
commit fe7942e
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/elasticgit/commands/tests/test_avro.py b/elasticgit/commands/tests/test_avro.py
@@ -400,7 +400,9 @@ def assertBinaryRoundTrip(self, model_class, data):
         with DataFileWriter(fp, DatumWriter(), schema) as writer:
             writer.append(dict(model))
 
-        with DataFileReader(open(file_name, 'rb'), DatumReader()) as reader:
+        with DataFileReader(
+                open(file_name, 'rb'),
+                DatumReader(readers_schema=schema)) as reader:
             [row] = reader
             self.assertEqual(model, model_class(row))