def foo_record_test(self):
     src = classes_source([FooRecord], indent=" " * 4)
     self.assertEqual(
         src,
         self.correct,
         msg='Incorrect definition:\n"""\n{0}"""\nShould have been:\n"""\n{1}"""'.format(src, self.correct),
     )
 def foo_record_test(self):
     src = classes_source([FooRecord], indent=" " * 4)
     self.assertEqual(
         src,
         self.correct,
         msg=
         "Incorrect definition:\n\"\"\"\n{0}\"\"\"\nShould have been:\n\"\"\"\n{1}\"\"\""
         .format(src, self.correct))
    def test_implicit_inclusion(self):
        src = classes_source([Parent])

        self.assertEquals(
            src,
            self.correct,
            msg="Incorrect definition:\n\"\"\"\n{0}\"\"\"\nShould have been:\n\"\"\"\n{1}\"\"\"".format(src, self.correct)
        )
 def foo_record_test(self):
     src = classes_source([FooRecord], indent=" " * 4)
     correct = (
         "class FooRecord(pyschema.Record):\n"
         "    # WARNING: This class was generated by pyschema.to_python_source\n"
         "    # there is a risk that any modification made to this class will be overwritten\n"
         "    _namespace = \'my.foo.bar\'\n"
         "    field_1 = Text(nullable=True, default=None)\n"
         "    bar = Integer(nullable=True, default=None)\n"
     )
     self.assertEqual(
         src,
         correct,
         msg="Incorrect definition:\n\"\"\"\n{0}\"\"\"\nShould have been:\n\"\"\"\n{1}\"\"\"".format(src, correct)
     )