def test_get_test_print_should_return_at_least_expected_classes(self): tpa = TestPrintAPI() name = "Half Vase With A Twist" height = 10 width = 10 layer_height = 0.1 speed = 99 result = tpa.get_test_print(name, height, width, layer_height, speed) self.assertEqual(type(result), HalfVaseTestGenerator) self.assertEqual(height, result._height) self.assertEqual(width / 2.0, result._max_radius) self.assertEqual(layer_height, result._layer_height) self.assertEqual(speed, result._speed)